aiagents4pharma 1.26.0__py3-none-any.whl → 1.26.1__py3-none-any.whl

aiagents4pharma/talk2aiagents4pharma/agents/main_agent.py

@@ -22,10 +22,11 @@ def get_app(uniq_id, llm_model: BaseChatModel):
     '''
     This function returns the langraph app.
     '''
-    if llm_model.model_name == 'gpt-4o-mini':
-        llm_model = ChatOpenAI(model='gpt-4o-mini',
-                               temperature=0,
-                               model_kwargs={"parallel_tool_calls": False})
+    if hasattr(llm_model, 'model_name'):
+        if llm_model.model_name == 'gpt-4o-mini':
+            llm_model = ChatOpenAI(model='gpt-4o-mini',
+                                temperature=0,
+                                model_kwargs={"parallel_tool_calls": False})
     # Load hydra configuration
     logger.log(logging.INFO, "Launching AIAgents4Pharma_Agent with thread_id %s", uniq_id)
     with hydra.initialize(version_base=None, config_path="../configs"):
@@ -57,7 +58,7 @@ def get_app(uniq_id, llm_model: BaseChatModel):
         # Full history is needed to extract
         # the tool artifacts
         output_mode="full_history",
-        add_handoff_back_messages=False,
+        add_handoff_back_messages=True,
         prompt=system_prompt
     )
 

{aiagents4pharma-1.26.0.dist-info → aiagents4pharma-1.26.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.26.0
+Version: 1.26.1
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D.
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{aiagents4pharma-1.26.0.dist-info → aiagents4pharma-1.26.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 aiagents4pharma/__init__.py,sha256=7xnvthMuxYurECWvyb0_yaPJ18SsqvmkKPTCxgbQlNQ,186
 aiagents4pharma/talk2aiagents4pharma/__init__.py,sha256=KOPe8cZ-ROQ6EGn8FeejRFUPokayKRMTgiXFyOpZGoA,122
 aiagents4pharma/talk2aiagents4pharma/agents/__init__.py,sha256=Cc1RitlLGzJ5d_dCSUdguZH417nlJux1qVCVS2M9t30,129
-aiagents4pharma/talk2aiagents4pharma/agents/main_agent.py,sha256=s0mrOk4DuAg7IkzB1YGyl_8wjxF2T_ReZENZirvjr94,2810
+aiagents4pharma/talk2aiagents4pharma/agents/main_agent.py,sha256=gnXBvvi3t8DjUtXImqgVJ9Rz7RqnDeHNTCMFiPo6bvc,2860
 aiagents4pharma/talk2aiagents4pharma/configs/__init__.py,sha256=5ah__-8XyRblwT0U1ByRigNjt_GyCheu7zce4aM-eZE,68
 aiagents4pharma/talk2aiagents4pharma/configs/config.yaml,sha256=VnbMbVSYfCh68cHZ0JLu00UjOUmapejN3EsN3lnBXtU,51
 aiagents4pharma/talk2aiagents4pharma/configs/agents/__init__.py,sha256=zrJcq-4m0YUKfSlRGC8KzBmEooaASKuL_Y75yDp-ZoA,72
@@ -186,8 +186,8 @@ aiagents4pharma/talk2scholars/tools/s2/search.py,sha256=i5KMFJWK31CjYtVT1McJpLzg
 aiagents4pharma/talk2scholars/tools/s2/single_paper_rec.py,sha256=7PoZfcstxDThWX6NYOgxN_9M_nwgMPAALch8OmjraVY,5568
 aiagents4pharma/talk2scholars/tools/zotero/__init__.py,sha256=1UW4r5ECvAwYpo1Fjf7lQPO--M8I85baYCHocFOAq4M,53
 aiagents4pharma/talk2scholars/tools/zotero/zotero_read.py,sha256=NJ65fAJ4u2Zq15uvEajVOhI4QnNvyqA6FHPaEDqvMw0,4321
-aiagents4pharma-1.26.0.dist-info/LICENSE,sha256=IcIbyB1Hyk5ZDah03VNQvJkbNk2hkBCDqQ8qtnCvB4Q,1077
-aiagents4pharma-1.26.0.dist-info/METADATA,sha256=qsXeD9uF6EQNZyWTWhm1T9YmKx286liyVn-W5CdIxCM,12585
-aiagents4pharma-1.26.0.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-aiagents4pharma-1.26.0.dist-info/top_level.txt,sha256=-AH8rMmrSnJtq7HaAObS78UU-cTCwvX660dSxeM7a0A,16
-aiagents4pharma-1.26.0.dist-info/RECORD,,
+aiagents4pharma-1.26.1.dist-info/LICENSE,sha256=IcIbyB1Hyk5ZDah03VNQvJkbNk2hkBCDqQ8qtnCvB4Q,1077
+aiagents4pharma-1.26.1.dist-info/METADATA,sha256=q4jfiTPl4koFlScZw3k22S_iIIr9w7SV-tk0aEftH-Q,12585
+aiagents4pharma-1.26.1.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+aiagents4pharma-1.26.1.dist-info/top_level.txt,sha256=-AH8rMmrSnJtq7HaAObS78UU-cTCwvX660dSxeM7a0A,16
+aiagents4pharma-1.26.1.dist-info/RECORD,,