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

fyodorov_llm_agents/agents/agent.py

@@ -5,7 +5,7 @@ import queue
 import threading
 import json
 import yaml
-from pydantic import BaseModel
+from pydantic import BaseModel, HttpUrl
 from openai import OpenAI as oai
 import litellm
 from fyodorov_llm_agents.tools.tool import Tool
@@ -15,7 +15,8 @@ MAX_DESCRIPTION_LENGTH = 280
 VALID_CHARACTERS_REGEX = r'^[a-zA-Z0-9\s.,!?:;\'"-_]+$'
 
 class Agent(BaseModel):
-    api_key: str = None
+    api_key: str | None = None
+    api_url: HttpUrl | None = None
     tools: [Tool] = []
     rag: [] = []
     model: str | None = None
@@ -76,10 +77,14 @@ class Agent(BaseModel):
         model = self.model
         # Set environmental variable
         if self.api_key.startswith('sk-'):
-            os.environ["OPENAI_API_KEY"] = self.api_key
-        else:
+            litellm.openai_key = self.api_key
+            self.api_url = "https://api.openai.com/v1"
+        elif self.api_key and self.api_key != '':
             model = 'mistral/'+self.model
-            os.environ["MISTRAL_API_KEY"] = self.api_key
+            litellm.mistral_key = self.api_key
+            litellm.api_base = "https://api.mistral.ai/v1"
+        else:
+            litellm.api_base = self.api_url
 
         messages: [] = [
             {"content": self.prompt, "role": "system"},

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

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

@@ -0,0 +1,7 @@
+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,,

{fyodorov_llm_agents-0.2.118.dist-info → fyodorov_llm_agents-0.2.122.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: bdist_wheel (0.43.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

fyodorov_llm_agents-0.2.118.dist-info/RECORD

@@ -1,7 +0,0 @@
-fyodorov_llm_agents/agents/agent.py,sha256=q-HVNdt4OvzB_FWxjgBxd_gG2ljJR7ZauVSysG-9CwM,5908
-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.118.dist-info/METADATA,sha256=HC4M0t4b60EaZSVf3ghDZu7t6dGyUgdEE52MpDxHQYQ,552
-fyodorov_llm_agents-0.2.118.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-fyodorov_llm_agents-0.2.118.dist-info/top_level.txt,sha256=4QOslsBp8Gh7ng25DceA7fHp4KguTIdAxwURz97gH-g,20
-fyodorov_llm_agents-0.2.118.dist-info/RECORD,,