fyodorov-llm-agents 0.2.122__py3-none-any.whl → 0.2.124__py3-none-any.whl

fyodorov_llm_agents/agents/agent.py

@@ -74,17 +74,16 @@ class Agent(BaseModel):
         }
 
     def call_with_fn_calling(self, input: str = "", history = []) -> dict:
+        litellm.set_verbose = True
         model = self.model
         # Set environmental variable
         if self.api_key.startswith('sk-'):
-            litellm.openai_key = self.api_key
+            os.environ["OPENAI_API_KEY"] = self.api_key
             self.api_url = "https://api.openai.com/v1"
         elif self.api_key and self.api_key != '':
             model = 'mistral/'+self.model
-            litellm.mistral_key = self.api_key
-            litellm.api_base = "https://api.mistral.ai/v1"
-        else:
-            litellm.api_base = self.api_url
+            os.environ["MISTRAL_API_KEY"] = self.api_key
+            self.api_url = "https://api.mistral.ai/v1"
 
         messages: [] = [
             {"content": self.prompt, "role": "system"},
@@ -95,10 +94,10 @@ class Agent(BaseModel):
         tools = [tool.get_function() for tool in self.tools]
         if tools and litellm.supports_function_calling(model=model):
             print(f"calling litellm with model {self.model}, tools: {tools}, messages: {messages}, max_retries: 0, history: {history}")
-            response = litellm.completion(model=model, messages=messages, max_retries=0, tools=tools, tool_choice="auto")
+            response = litellm.completion(model=model, messages=messages, max_retries=0, tools=tools, tool_choice="auto", base_url=self.api_url)
         else:
             print(f"calling litellm with model {self.model}, messages: {messages}, max_retries: 0, history: {history}")
-            response = litellm.completion(model=model, messages=messages, max_retries=0)
+            response = litellm.completion(model=model, messages=messages, max_retries=0, base_url=self.api_url)
         print(f"Response: {response}")
         tool_calls = []
         if hasattr(response, 'tool_calls'):

{fyodorov_llm_agents-0.2.122.dist-info → fyodorov_llm_agents-0.2.124.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fyodorov_llm_agents
-Version: 0.2.122
+Version: 0.2.124
 Summary: LLM agents for the Fyodorov AI suite
 Author-email: Daniel Ransom <02masseur.alibis@icloud.com>
 Project-URL: Homepage, https://github.com/FyodorovAI/fyodorov-llm-agents

fyodorov_llm_agents-0.2.124.dist-info/RECORD

@@ -0,0 +1,7 @@
+fyodorov_llm_agents/agents/agent.py,sha256=bDztxFseKX-C3IlMFwSEVIrXUWppnNb_b4Mita8iUe4,6186
+fyodorov_llm_agents/agents/openai.py,sha256=FA5RS7yn3JwvFA8PXju60XSYC_2oUZFNgBUzeIYtGv0,1154
+fyodorov_llm_agents/tools/tool.py,sha256=HyOk0X_3XE23sa8J-8UZx657tJ0sxwZWMbA4OPxXU6E,7940
+fyodorov_llm_agents-0.2.124.dist-info/METADATA,sha256=UcAHX6orrqB1uzMwbaq45OTc3kgit7_8bT4DqDovZ5M,552
+fyodorov_llm_agents-0.2.124.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+fyodorov_llm_agents-0.2.124.dist-info/top_level.txt,sha256=4QOslsBp8Gh7ng25DceA7fHp4KguTIdAxwURz97gH-g,20
+fyodorov_llm_agents-0.2.124.dist-info/RECORD,,

fyodorov_llm_agents-0.2.122.dist-info/RECORD

@@ -1,7 +0,0 @@
-fyodorov_llm_agents/agents/agent.py,sha256=h0jhRCrCknx1ORGxWWD1b9zWV3coc5XlFUtimL5235E,6147
-fyodorov_llm_agents/agents/openai.py,sha256=FA5RS7yn3JwvFA8PXju60XSYC_2oUZFNgBUzeIYtGv0,1154
-fyodorov_llm_agents/tools/tool.py,sha256=HyOk0X_3XE23sa8J-8UZx657tJ0sxwZWMbA4OPxXU6E,7940
-fyodorov_llm_agents-0.2.122.dist-info/METADATA,sha256=PJiOquJA5BLNqZkbGS038vanXsQ3jef-uqEIyYcQ3l4,552
-fyodorov_llm_agents-0.2.122.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-fyodorov_llm_agents-0.2.122.dist-info/top_level.txt,sha256=4QOslsBp8Gh7ng25DceA7fHp4KguTIdAxwURz97gH-g,20
-fyodorov_llm_agents-0.2.122.dist-info/RECORD,,