versionhq 1.2.1.18__py3-none-any.whl → 1.2.1.20__py3-none-any.whl

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/_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/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/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/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/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/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.dist-info → versionhq-1.2.1.20.dist-info}/METADATA

@@ -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.dist-info → versionhq-1.2.1.20.dist-info}/RECORD

@@ -1,19 +1,19 @@
-versionhq/__init__.py,sha256=I1z2OKVSflVHMiWtAsm5qE2-pmpxTtiMQhcytzInHas,2892
+versionhq/__init__.py,sha256=swg84hsZ4rVrvtBLOAN8LcDfiZRR1jJ2sZR9-_iTedg,2892
 versionhq/_utils/__init__.py,sha256=dzoZr4cBlh-2QZuPzTdehPUCe9lP1dmRtauD7qTjUaA,158
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
-versionhq/_utils/logger.py,sha256=zgogTwAY-ujDLrdryAKhdtoaNe1nOFajmEN0V8aMR34,3155
+versionhq/_utils/logger.py,sha256=iHxGjm3BvUo5dHKLU88_pc0Z45wzSHOjyJGQkb7OADk,3255
 versionhq/_utils/process_config.py,sha256=YTGY_erW335RfceQfzS18YAqq-AAb-iSvKSjN7noD2E,782
 versionhq/_utils/usage_metrics.py,sha256=NXF18dn5NNvGK7EsQ4AAghpR8ppYOjMx6ABenLLHnmM,1066
 versionhq/_utils/vars.py,sha256=bZ5Dx_bFKlt3hi4-NNGXqdk7B23If_WaTIju2fiTyPQ,57
 versionhq/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/agent/inhouse_agents.py,sha256=vupO1viYqVb7sKohIE1zThu6JArhh5JLo5LBeSnh0kM,2534
-versionhq/agent/model.py,sha256=acAFIDmCbSVbMIf0Qlp9lzhdF0f_201havj3RnYj0xw,25661
+versionhq/agent/model.py,sha256=2sFRDYk47yzgzHKaS0JGobJ-4q96UV4uz11BAsF_RXA,26535
 versionhq/agent/parser.py,sha256=riG0dkdQCxH7uJ0AbdVdg7WvL0BXhUgJht0VtQvxJBc,4082
 versionhq/agent/rpm_controller.py,sha256=grezIxyBci_lDlwAlgWFRyR5KOocXeOhYkgN02dNFNE,2360
 versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3VEXUreOcKs,532
 versionhq/agent/TEMPLATES/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/agent_network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/agent_network/formation.py,sha256=QHLbv4XgbmMEQpdoHGrV6_CQOW3kIe7Jp32G0HgpA90,7418
+versionhq/agent_network/formation.py,sha256=7iGw20Dj2sFLAho6yfrdmonAwFcxINBDGFr2RU-Qz3s,7505
 versionhq/agent_network/model.py,sha256=hjtYIopAN52nStcM6TlV0b6ulRMrmzKH7jIkzNmZHDE,19265
 versionhq/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,7 +26,7 @@ versionhq/clients/workflow/model.py,sha256=fqmTx8Y6P6i-sK045ENO88GDbf4WYY4hUA26R
 versionhq/knowledge/__init__.py,sha256=qW7IgssTA4_bFFV9ziOcYRfGjlq1c8bkb-HnfWknpuQ,567
 versionhq/knowledge/_utils.py,sha256=YWRF8U533cfZes_gZqUvdj-K24MD2ri1R0gjc_aPYyc,402
 versionhq/knowledge/embedding.py,sha256=KfHc__1THxb5jrg1EMrF-v944RDuIr2hE0l-MtM3Bp0,6826
-versionhq/knowledge/model.py,sha256=w29mrJv1kiznCh4P4yJMUQxIuyRw1Sk0XYtBXzCxaG4,1786
+versionhq/knowledge/model.py,sha256=ixH8n5kLtJEp1nPAFYA0piYm-n0nnFDtWFp0r9YEVAs,1787
 versionhq/knowledge/source.py,sha256=-hEUPtJUHHMx4rUKtiHl19J8xAMw-WVBw34zwa2jZ08,13630
 versionhq/knowledge/source_docling.py,sha256=mg7bgvKePHn2LlA_XzSFCbS0zOo9xfu_aNOf5cEo6c4,5421
 versionhq/knowledge/storage.py,sha256=7oxCg3W9mFjYH1YmuH9kFtTbNxquzYFjuUjd_TlsB9E,8170
@@ -39,8 +39,8 @@ versionhq/memory/model.py,sha256=MPO8dDP5eAuk9td6bMOq5j2huLzCADFJDrTujHhcWQY,815
 versionhq/storage/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/storage/base.py,sha256=p-Jas0fXQan_qotnRD6seQxrT2lj-uw9-SmHQhdppcs,355
 versionhq/storage/ltm_sqlite_storage.py,sha256=wdUiuwHfJocdk0UGqyrdU4S5Nae1rgsoRNu3LWmGFcI,3951
-versionhq/storage/mem0_storage.py,sha256=Nl0GlCNftZGTVxX-9DP5n_pN4QRxeHAhAPakCc0arBg,3819
-versionhq/storage/rag_storage.py,sha256=ScWC0vH327vnGw8UGscAOoIfqrq3mhvXT3vEKzHZJts,7441
+versionhq/storage/mem0_storage.py,sha256=ZY8MELBWaINRv9YuRW5MxH7dj2cII-L0i3xSD6o1-2M,3781
+versionhq/storage/rag_storage.py,sha256=bS2eE874obarYl-4hT6ZWYWTRsqtfuGpKgKzERmM6Uo,7433
 versionhq/storage/task_output_storage.py,sha256=E1t_Fkt78dPYIOl3MP7LfQ8oGtjlzxBuSNq_8ZXKho8,4573
 versionhq/storage/utils.py,sha256=ByYXPoEIGJYLUqz-DWjbCAnneNrH1otiYbp12SCILpM,747
 versionhq/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -61,8 +61,8 @@ versionhq/tool/composio_tool_vars.py,sha256=FvBuEXsOQUYnN7RTFxT20kAkiEYkxWKkiVtg
 versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1205
 versionhq/tool/model.py,sha256=PO4zNWBZcJhYVur381YL1dy6zqurio2jWjtbxOxZMGI,12194
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.2.1.18.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.1.18.dist-info/METADATA,sha256=NvKh5tYpJbFp0SHQaukksKnHOCrzO2SBGJfO0rJlTZM,22033
-versionhq-1.2.1.18.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-versionhq-1.2.1.18.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.1.18.dist-info/RECORD,,
+versionhq-1.2.1.20.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.1.20.dist-info/METADATA,sha256=KzJg8vQM1tOkvS9seXV_wCgcgN5l_bG0xw9KcvxWVI0,22033
+versionhq-1.2.1.20.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+versionhq-1.2.1.20.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.1.20.dist-info/RECORD,,