versionhq 1.2.1.18__tar.gz → 1.2.1.20__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.1.18
+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.18 → versionhq-1.2.1.20}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.1.18"
+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.18 → 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.18"
+__version__ = "1.2.1.20"
 __all__ = [
     "Agent",
 

{versionhq-1.2.1.18 → 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.18 → 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(' ', '_')}"
+                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.18 → versionhq-1.2.1.20}/src/versionhq/agent_network/formation.py

@@ -9,6 +9,9 @@ from versionhq.agent_network.model import AgentNetwork, Member, Formation
 from versionhq.agent.inhouse_agents  import vhq_formation_planner
 from versionhq._utils import Logger
 
+import chromadb
+chromadb.api.client.SharedSystemClient.clear_system_cache()
+
 
 def form_agent_network(
         task: str,
@@ -85,7 +88,7 @@ def form_agent_network(
     members = []
     leader = str(res.pydantic.leader_agent) if res.pydantic else str(res.json_dict["leader_agent"])
 
-    created_agents = [Agent(role=item, goal=item) for item in res.pydantic.agent_roles]
+    created_agents = [Agent(role=str(item), goal=str(item)) for item in res.pydantic.agent_roles]
     created_tasks = []
 
     if res.pydantic:

{versionhq-1.2.1.18 → versionhq-1.2.1.20}/src/versionhq/knowledge/model.py

@@ -9,6 +9,7 @@ class Knowledge(BaseModel):
     """
     Knowlede class for collection of sources and setup for the vector store to query relevant context.
     """
+
     collection_name: Optional[str] = None
     sources: List[BaseKnowledgeSource] = Field(default_factory=list)
     storage: KnowledgeStorage = Field(default_factory=KnowledgeStorage)

{versionhq-1.2.1.18 → 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.18 → 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
@@ -129,23 +127,23 @@ class RAGStorage(BaseRAGStorage):
         from chromadb.config import Settings
 
         self._set_embedder_config()
-        chroma_client = chromadb.PersistentClient(
-            path=self.path if self.path else self.storage_file_name,
-            settings=Settings(allow_reset=self.allow_reset),
-        )
-        self.app = chroma_client
-
         try:
+            chroma_client = chromadb.PersistentClient(
+                path=self.path if self.path else self.storage_file_name,
+                settings=Settings(allow_reset=self.allow_reset),
+            )
+            self.app = chroma_client
             self.collection = self.app.get_collection(name=self.type, embedding_function=self.embedder_config)
         except Exception:
-            self.collection = self.app.create_collection(name=self.type, embedding_function=self.embedder_config)
+            if self.app:
+                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.18 → 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.18
+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.18 → versionhq-1.2.1.20}/tests/agent/agent_test.py

@@ -244,10 +244,12 @@ def test_agent_with_knowledge_sources():
 
     agent = Agent(role="Information Agent", goal="Provide information based on knowledge sources", knowledge_sources=knowledge_sources)
 
-    assert agent._knowledge.collection_name == f"{agent.role.replace(' ', '_')}"
+    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 and  agent._knowledge.storage.app is not None and agent._knowledge.storage.collection_name is not None
+    assert agent._knowledge.storage and agent._knowledge.storage.embedding_function
+    assert agent._knowledge.storage.app is not None
+    assert agent._knowledge.storage.collection_name is not None
 
     task = Task(description="Answer the following question: What is Kuriko's favorite color?")
 

{versionhq-1.2.1.18 → 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.18 → 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.18 → versionhq-1.2.1.20}/uv.lock

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