aiagents4pharma 1.26.0__tar.gz → 1.26.2__tar.gz

{aiagents4pharma-1.26.0 → aiagents4pharma-1.26.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.26.0
+Version: 1.26.2
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D.
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -91,6 +91,7 @@ Our toolkit currently consists of the following agents:
 ![Python Version from PEP 621 TOML](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2FVirtualPatientEngine%2FAIAgents4Pharma%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)
 
 ### Installation
+_Please use version 1.26.2 or later for better support with NVIDIA NIM models._
 
 #### Option 1: PyPI
 

{aiagents4pharma-1.26.0 → aiagents4pharma-1.26.2}/README.md

@@ -26,6 +26,7 @@ Our toolkit currently consists of the following agents:
 ![Python Version from PEP 621 TOML](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2FVirtualPatientEngine%2FAIAgents4Pharma%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)
 
 ### Installation
+_Please use version 1.26.2 or later for better support with NVIDIA NIM models._
 
 #### Option 1: PyPI
 

{aiagents4pharma-1.26.0 → aiagents4pharma-1.26.2}/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.2/aiagents4pharma/talk2aiagents4pharma/configs/agents/main_agent/default.yaml

@@ -0,0 +1,16 @@
+_target_: agents.main_agent.get_app
+system_prompt: >
+  You are Talk2AIAgents4Pharma agent.
+  You are managing a team of the following 2 agents:
+   
+  1. Talk2Biomodels (T2B) agent: This agent can search and 
+  operate on mathematical models of biological systems. This
+  agent can also query an uploaded document/pdf/article. 
+   
+  2. Talk2KnowledgeGraphs (T2KG) agent: This agent can
+  reason over a knowledge graph of biological entities
+  and their relationships to answer subjective questions.
+  Do not call this agent for objective questions or questions
+  related to the mathematical models of biological systems.
+
+  Below mentioned are the prompts for each agent:

{aiagents4pharma-1.26.0 → aiagents4pharma-1.26.2}/aiagents4pharma.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: aiagents4pharma
-Version: 1.26.0
+Version: 1.26.2
 Summary: AI Agents for drug discovery, drug development, and other pharmaceutical R&D.
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -91,6 +91,7 @@ Our toolkit currently consists of the following agents:
 ![Python Version from PEP 621 TOML](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2FVirtualPatientEngine%2FAIAgents4Pharma%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)
 
 ### Installation
+_Please use version 1.26.2 or later for better support with NVIDIA NIM models._
 
 #### Option 1: PyPI
 

aiagents4pharma-1.26.2/release_version.txt

@@ -0,0 +1 @@
+v1.26.2

aiagents4pharma-1.26.0/aiagents4pharma/talk2aiagents4pharma/configs/agents/main_agent/default.yaml

@@ -1,12 +0,0 @@
-_target_: agents.main_agent.get_app
-system_prompt: >
-  You are Talk2AIAgents4Pharma agent.
-  You are managing a team of the following 2 agents:
-   
-  1. Talk2Biomodels (T2B) agent: This agent can operate
-  on mathematical models of biological systems. This
-  agent can also query an uploaded document/pdf/article. 
-   
-  2. Talk2KnowledgeGraphs (T2KG) agent: This agent can
-  reason over a knowledge graph of biological entities
-  and their relationships.

aiagents4pharma-1.26.0/release_version.txt

@@ -1 +0,0 @@
-v1.26.0