fyodorov-llm-agents 0.0.92__py3-none-any.whl → 0.0.93__py3-none-any.whl

fyodorov_llm_agents/agents/base_agent.py

@@ -6,13 +6,71 @@ import threading
 import json
 from openai import OpenAI as oai
 
+MAX_NAME_LENGTH = 80
+MAX_DESCRIPTION_LENGTH = 280
+VALID_CHARACTERS_REGEX = r'^[a-zA-Z0-9\s.,!?:;\'"-]+$'
+
 class BaseAgent():
-    def __init__(self, api_key, tools: [Tool] = [], rag: [] = [], model: str = "gpt-3.5-turbo"):
+    provider_id: str
+    name_for_human: str
+    description_for_human: str
+    prompt: str
+
+    def __init__(
+        self, 
+        api_key,
+        tools: [Tool] = [],
+        rag: [] = [],
+        model: str = "gpt-3.5-turbo",
+        name_for_human: str = "My Agent",
+        description_for_human: str = "My Agent",
+        prompt: str = "My Prompt",
+        prompt_size: int = 10000,
+        temperature: float = 0.8,
+    ):
         self.api_key = api_key
         self.client = oai(api_key=api_key)
         self.tools = tools
         self.rag = rag
         self.model = model
+        self.name_for_human = name_for_human
+        self.description_for_human = description_for_human
+        self.prompt = prompt
+        self.validate()
+
+    def validate(self):
+        BaseAgent.validate_name_for_human(self.name_for_human)
+        BaseAgent.validate_description_for_human(self.description_for_human)
+        BaseAgent.validate_prompt(self.prompt, self.prompt_size)
+
+    @staticmethod
+    def validate_name_for_human(name_for_human: str) -> str:
+        if not name_for_human:
+            raise ValueError('Name for human is required')
+        if len(name_for_human) > MAX_NAME_LENGTH:
+            raise ValueError('Name for human exceeds maximum length')
+        if not re.match(VALID_CHARACTERS_REGEX, name_for_human):
+            raise ValueError('Name for human contains invalid characters')
+        return name_for_human
+
+    @staticmethod
+    def validate_description_for_human(description_for_human: str) -> str:
+        if not description_for_human:
+            raise ValueError('Description for human is required')
+        if len(description_for_human) > MAX_DESCRIPTION_LENGTH:
+            raise ValueError('Description for human exceeds maximum length')
+        if not re.match(VALID_CHARACTERS_REGEX, description_for_human):
+            raise ValueError('Description for human contains invalid characters')
+        return description_for_human
+
+    @staticmethod
+    def validate_prompt(prompt: str, prompt_size: int) -> str:
+        if not prompt:
+            raise ValueError('Prompt is required')
+        if len(prompt) > prompt_size:
+            raise ValueError('Prompt exceeds maximum length')
+        return prompt
+
 
     def call(self, prompt: str = "", input: str = "", temperature: float = 0.0):
         print(f"[OpenAI] Calling OpenAI with prompt: {prompt}, input: {input}, temperature: {temperature}")
@@ -154,6 +212,25 @@ class BaseAgent():
                 print(f"type of result: {type(result)}")
                 yield result.encode()
     
+    @staticmethod
+    def from_yaml(yaml_str: str):
+        """Instantiate BaseAgent from YAML."""
+        if not yaml_str:
+            raise ValueError('YAML string is required')
+        agent_dict = yaml.safe_load(yaml_str)
+        agent = cls(
+            api_key=agent_dict["api_key"],
+            model=agent_dict["model"],
+            tools=agent_dict["tools"],
+            rag=agent_dict["rag"],
+            name_for_human=agent_dict["name_for_human"],
+            description_for_human=agent_dict["description_for_human"],
+            prompt=agent_dict["prompt"],
+            temperature=agent_dict["temperature"],
+            prompt_size=agent_dict["prompt_size"],
+        )
+        return agent
+    
     def add_tool(self, tool: Tool):
         self.tools.append(tool)
 

{fyodorov_llm_agents-0.0.92.dist-info → fyodorov_llm_agents-0.0.93.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fyodorov_llm_agents
-Version: 0.0.92
+Version: 0.0.93
 Summary: Library for creating LLM-based agents with tools and RAG
 Author-email: Daniel Ransom <02masseur.alibis@icloud.com>
 Project-URL: Homepage, https://github.com/FyodorovAI/fyodorov-llm-agents

{fyodorov_llm_agents-0.0.92.dist-info → fyodorov_llm_agents-0.0.93.dist-info}/RECORD

@@ -1,12 +1,12 @@
 fyodorov_llm_agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fyodorov_llm_agents/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-fyodorov_llm_agents/agents/base_agent.py,sha256=uEgjX95kQcImDCs_epTUuprmB0sjWMEODX2DOFtzUgc,11142
+fyodorov_llm_agents/agents/base_agent.py,sha256=YTUV3bokCxm_Cf2FxYdLHOjP4Gj-8xNN8pAl30NXz9Q,13949
 fyodorov_llm_agents/agents/openai.py,sha256=yAGV3pcFcKfwHR1zuIiv1hmmWOUjDfrJloUGECciOlw,1167
 fyodorov_llm_agents/prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fyodorov_llm_agents/prompts/prompt.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fyodorov_llm_agents/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fyodorov_llm_agents/tools/tool.py,sha256=7N_v1I0lRkDrT1qTYlEgcxzbrLopH-uegkyybSV8BEk,6976
-fyodorov_llm_agents-0.0.92.dist-info/METADATA,sha256=8Jy4_JPzjFVohir_gNFLKdYPl6cDr_WBqZ73HmNfSxk,2001
-fyodorov_llm_agents-0.0.92.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-fyodorov_llm_agents-0.0.92.dist-info/top_level.txt,sha256=4QOslsBp8Gh7ng25DceA7fHp4KguTIdAxwURz97gH-g,20
-fyodorov_llm_agents-0.0.92.dist-info/RECORD,,
+fyodorov_llm_agents-0.0.93.dist-info/METADATA,sha256=uIM52gkBYD59OE7hUVOER6xr-rnButDgHetvHMA0wHM,2001
+fyodorov_llm_agents-0.0.93.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+fyodorov_llm_agents-0.0.93.dist-info/top_level.txt,sha256=4QOslsBp8Gh7ng25DceA7fHp4KguTIdAxwURz97gH-g,20
+fyodorov_llm_agents-0.0.93.dist-info/RECORD,,