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

fyodorov_llm_agents/agents/agent.py

@@ -77,14 +77,12 @@ class Agent(BaseModel):
         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 +93,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", api_base=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, api_base=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.123.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fyodorov_llm_agents
-Version: 0.2.122
+Version: 0.2.123
 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.123.dist-info/RECORD

@@ -0,0 +1,7 @@
+fyodorov_llm_agents/agents/agent.py,sha256=MrRfaK6ghYJC9EpJjAWHo_sIrsHKZn98ATRwxDDlrMA,6151
+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.123.dist-info/METADATA,sha256=iaBS0UaG2bbZZxESVODWqaPj3IdvtjR4QqkHyrInOxo,552
+fyodorov_llm_agents-0.2.123.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+fyodorov_llm_agents-0.2.123.dist-info/top_level.txt,sha256=4QOslsBp8Gh7ng25DceA7fHp4KguTIdAxwURz97gH-g,20
+fyodorov_llm_agents-0.2.123.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,,