versionhq 1.2.1.19__tar.gz → 1.2.1.20__tar.gz

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.19
+Version: 1.2.1.20
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.19"
+version = "1.2.1.20"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
 description = "An agentic orchestration framework for building agent networks that handle task automation."
 readme = "README.md"

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/src/versionhq/__init__.py

@@ -31,7 +31,7 @@ from versionhq.agent_network.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.1.19"
+__version__ = "1.2.1.20"
 __all__ = [
     "Agent",
 

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/src/versionhq/_utils/logger.py

@@ -44,6 +44,7 @@ class Logger(BaseModel):
 
     verbose: bool = Field(default=True)
     info_file_save: bool = Field(default=False, description="whether to save INFO logs")
+    filename: str = Field(default=None)
     _printer: Printer = PrivateAttr(default_factory=Printer)
 
 
@@ -52,7 +53,7 @@ class Logger(BaseModel):
             timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
             self._printer.print(f"\n{timestamp} - versionHQ [{level.upper()}]: {message}", color=color)
 
-        self._save(level=level, message=message)
+        self._save(level=level, message=message, filename=self.filename)
 
 
     def _save(self, level: str, message: str, filename: str = None):
@@ -74,7 +75,7 @@ class Logger(BaseModel):
         cwd = Path.cwd()
         log_file_dir = f"{cwd}/.logs"
         os.makedirs(log_file_dir, exist_ok=True)
-        filename = filename if filename else datetime.now().strftime('%H_%M_%S_%d_%m_%Y')
+        filename = filename if filename else self.filename if self.filename else datetime.now().strftime('%H_%M_%S_%d_%m_%Y')
         abs_dir = f"{log_file_dir}/{filename}.log"
 
         logging.basicConfig(filename=abs_dir, filemode='w', level=logging_level)

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/src/versionhq/agent/model.py

@@ -64,6 +64,7 @@ class Agent(BaseModel):
     _request_within_rpm_limit: Any = PrivateAttr(default=None)
     _token_process: TokenProcess = PrivateAttr(default_factory=TokenProcess)
     _times_executed: int = PrivateAttr(default=0)
+    _logger_config: Dict[str, Any] = PrivateAttr(default=dict(verbose=True, info_file_save=True))
     config: Optional[Dict[str, Any]] = Field(default=None, exclude=True, description="values to add to the Agent class")
 
     id: UUID4 = Field(default_factory=uuid.uuid4, frozen=True)
@@ -172,7 +173,7 @@ class Agent(BaseModel):
                     tool_list.append(item)
 
                 else:
-                    Logger().log(level="error", message=f"Tool {str(item)} is missing a function.", color="red")
+                    Logger(**self.loger_config, filename=self.key).log(level="error", message=f"Tool {str(item)} is missing a function.", color="red")
                     raise PydanticCustomError("invalid_tool", f"The tool {str(item)} is missing a function.", {})
 
             self.tools = tool_list
@@ -222,7 +223,7 @@ class Agent(BaseModel):
 
         if self.knowledge_sources:
             try:
-                collection_name = f"{self.role.replace(' ', '_')}-{str(self.id)}"
+                collection_name = self.key
                 knowledge_sources = []
                 docling_fp, txt_fp, json_fp, excel_fp, csv_fp, pdf_fp = [], [], [], [], [], []
                 str_cont = ""
@@ -256,7 +257,7 @@ class Agent(BaseModel):
                 self._knowledge = Knowledge(sources=knowledge_sources, embedder_config=self.embedder_config, collection_name=collection_name)
 
             except:
-                Logger().log(level="warning", message="We cannot find the format for the source. Add BaseKnowledgeSource objects instead.", color="yellow")
+                Logger(**self._logger_config, filename=self.key).log(level="warning", message="We cannot find the format for the source. Add BaseKnowledgeSource objects instead.", color="yellow")
 
         return self
 
@@ -375,7 +376,7 @@ class Agent(BaseModel):
         """
 
         if not llm and not llm_config:
-            Logger().log(level="error", message="Missing llm or llm_config values to update", color="red")
+            Logger(**self._logger_config, filename=self.key).log(level="error", message="Missing llm or llm_config values to update", color="red")
             pass
 
         self.llm = llm
@@ -423,7 +424,7 @@ class Agent(BaseModel):
             if self._rpm_controller and self.max_rpm:
                 self._rpm_controller.check_or_wait()
 
-            Logger().log(level="info", message=f"Messages sent to the model: {messages}", color="blue")
+            Logger(**self._logger_config, filename=self.key).log(level="info", message=f"Messages sent to the model: {messages}", color="blue")
 
             if tool_res_as_final:
                 raw_response = self.func_calling_llm.call(messages=messages, tools=tools, tool_res_as_final=True)
@@ -433,11 +434,11 @@ class Agent(BaseModel):
                 task.tokens = self.llm._tokens
 
             task_execution_counter += 1
-            Logger().log(level="info", message=f"Agent response: {raw_response}", color="green")
+            Logger(**self._logger_config, filename=self.key).log(level="info", message=f"Agent response: {raw_response}", color="green")
             return raw_response
 
         except Exception as e:
-            Logger().log(level="error", message=f"An error occured. The agent will retry: {str(e)}", color="red")
+            Logger(**self._logger_config, filename=self.key).log(level="error", message=f"An error occured. The agent will retry: {str(e)}", color="red")
 
             while not raw_response and task_execution_counter <= self.max_retry_limit:
                 while (not raw_response or raw_response == "" or raw_response is None) and iterations < self.maxit:
@@ -449,11 +450,11 @@ class Agent(BaseModel):
                     iterations += 1
 
                 task_execution_counter += 1
-                Logger().log(level="info", message=f"Agent #{task_execution_counter} response: {raw_response}", color="green")
+                Logger(**self._logger_config, filename=self.key).log(level="info", message=f"Agent #{task_execution_counter} response: {raw_response}", color="green")
                 return raw_response
 
             if not raw_response:
-                Logger().log(level="error", message="Received None or empty response from the model", color="red")
+                Logger(**self._logger_config, filename=self.key).log(level="error", message="Received None or empty response from the model", color="red")
                 raise ValueError("Invalid response from LLM call - None or empty.")
 
 
@@ -463,7 +464,7 @@ class Agent(BaseModel):
         """
 
         if not kwargs:
-            Logger().log(level="error", message="Missing values to update", color="red")
+            Logger(**self._logger_config, filename=self.key).log(level="error", message="Missing values to update", color="red")
             return self
 
         for k, v in kwargs.items():
@@ -503,7 +504,7 @@ class Agent(BaseModel):
                     try:
                         setattr(self, k, v)
                     except Exception as e:
-                        Logger().log(level="error", message=f"Failed to update the field: {k} We'll skip it. Error: {str(e)}", color="red")
+                        Logger(**self._logger_config, filename=self.key).log(level="error", message=f"Failed to update the field: {k} We'll skip it. Error: {str(e)}", color="red")
                         pass
 
         return self
@@ -584,11 +585,11 @@ class Agent(BaseModel):
 
         except Exception as e:
             self._times_executed += 1
-            Logger().log(level="error", message=f"The agent failed to execute the task. Error: {str(e)}", color="red")
+            Logger(**self._logger_config, filename=self.key).log(level="error", message=f"The agent failed to execute the task. Error: {str(e)}", color="red")
             raw_response = self.execute_task(task, context, task_tools)
 
             if self._times_executed > self.max_retry_limit:
-                Logger().log(level="error", message=f"Max retry limit has exceeded.", color="red")
+                Logger(**self._logger_config, filename=self.key).log(level="error", message=f"Max retry limit has exceeded.", color="red")
                 raise e
 
         if self.max_rpm and self._rpm_controller:
@@ -597,6 +598,15 @@ class Agent(BaseModel):
         return raw_response
 
 
+    @property
+    def key(self):
+        """
+        A key to identify an agent. Used in storage, logging, and other recodings.
+        """
+        sanitized_role = self.role.lower().replace(" ", "-").replace("/", "").replace("{", "").replace("}", "").replace("\n", "")
+        return f"{str(self.id)}-{sanitized_role}"
+
+
     def __repr__(self):
         return f"Agent(role={self.role}, goal={self.goal}"
 

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/src/versionhq/storage/mem0_storage.py

@@ -40,11 +40,11 @@ class Mem0Storage(Storage):
             self.memory = MemoryClient(api_key=mem0_api_key)
 
 
-    def _sanitize_role(self, role: str) -> str:
-        """
-        Sanitizes agent roles to ensure valid directory names.
-        """
-        return role.replace("\n", "").replace(" ", "_").replace("/", "_")
+    # def _sanitize_role(self, role: str) -> str:
+    #     """
+    #     Sanitizes agent roles to ensure valid directory names.
+    #     """
+    #     return role.replace("\n", "").replace(" ", "_").replace("/", "_")
 
 
     def save(self, value: Dict[str, Any] | str, metadata: Dict[str, Any]) -> None:
@@ -104,6 +104,5 @@ class Mem0Storage(Storage):
 
     def _get_agent_name(self):
         agents = self.agents if self.agents else []
-        agents = [self._sanitize_role(agent.role) for agent in agents]
-        agents = "_".join(agents)
+        agents = "_".join([item.key for item in agents])
         return agents

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/src/versionhq/storage/rag_storage.py

@@ -52,15 +52,13 @@ class BaseRAGStorage(ABC):
 
     def _initialize_agents(self) -> str:
         if self.agents:
-            return "_".join(
-                [self._sanitize_role(agent.role) for agent in self.agents]
-            )
+            return "_".join([agent.key for agent in self.agents])
         return ""
 
-    @abstractmethod
-    def _sanitize_role(self, role: str) -> str:
-        """Sanitizes agent roles to ensure valid directory names."""
-        pass
+    # @abstractmethod
+    # def _sanitize_role(self, role: str) -> str:
+    #     """Sanitizes agent roles to ensure valid directory names."""
+    #     pass
 
     @abstractmethod
     def save(self, value: Any, metadata: Dict[str, Any]) -> None:
@@ -108,7 +106,7 @@ class RAGStorage(BaseRAGStorage):
     def __init__(self, type, allow_reset=True, embedder_config=None, agents=list(), path=None):
         super().__init__(type, allow_reset, embedder_config, agents)
         agents = agents
-        agents = [self._sanitize_role(agent.role) for agent in agents]
+        agents = [agent.key for agent in agents]
         agents = "_".join(agents)
 
         self.agents = agents
@@ -141,11 +139,11 @@ class RAGStorage(BaseRAGStorage):
                 self.collection = self.app.create_collection(name=self.type, embedding_function=self.embedder_config)
 
 
-    def _sanitize_role(self, role: str) -> str:
-        """
-        Sanitizes agent roles to ensure valid directory names.
-        """
-        return role.replace("\n", "").replace(" ", "_").replace("/", "_")
+    # def _sanitize_role(self, role: str) -> str:
+    #     """
+    #     Sanitizes agent roles to ensure valid directory names.
+    #     """
+    #     return role.replace("\n", "").replace(" ", "_").replace("/", "_")
 
 
     def _build_storage_file_name(self, type: str, file_name: str) -> str:

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/src/versionhq.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.19
+Version: 1.2.1.20
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/tests/agent/agent_test.py

@@ -244,7 +244,7 @@ def test_agent_with_knowledge_sources():
 
     agent = Agent(role="Information Agent", goal="Provide information based on knowledge sources", knowledge_sources=knowledge_sources)
 
-    assert  f"{agent.role.replace(' ', '_')}" in agent._knowledge.collection_name
+    assert agent._knowledge.collection_name == agent.key
     assert [isinstance(item, StringKnowledgeSource | DoclingSource) for item in agent.knowledge_sources]
     assert agent._knowledge.embedder_config == agent.embedder_config
     assert agent._knowledge.storage and agent._knowledge.storage.embedding_function

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/tests/formation_test.py

@@ -15,7 +15,7 @@ def test_minimum_inputs():
 
 
 
-def test_specific_formation():
+def _test_specific_formation():
     import versionhq as vhq
 
     formations_to_test = [

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/tests/usecase_test.py

@@ -52,6 +52,6 @@ def test_demo_agent_customization():
     )
 
     assert "gemini-2.0" in agent.llm.model
-    assert agent.llm.temperature == 1 and agent.llm.top_p == 0.1 and agent.llm.n==1 and agent.llm.stop == "test"
+    assert agent.llm.temperature == 1 and agent.llm.top_p == 0.1 and agent.llm.n == 1 and agent.llm.stop == "test"
     assert agent.knowledge_sources == ['https://business.linkedin.com', f'{current_path}/demo.csv',]
     assert agent.with_memory == True

{versionhq-1.2.1.19 → versionhq-1.2.1.20}/uv.lock

@@ -3982,7 +3982,7 @@ wheels = [
 
 [[package]]
 name = "versionhq"
-version = "1.2.1.18"
+version = "1.2.1.20"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },