versionhq 1.2.3.4__py3-none-any.whl → 1.2.3.6__py3-none-any.whl

versionhq/__init__.py

@@ -32,7 +32,7 @@ from versionhq.agent_network.formation import form_agent_network
 from versionhq.task_graph.draft import workflow
 
 
-__version__ = "1.2.3.4"
+__version__ = "1.2.3.6"
 __all__ = [
     "Agent",
 

versionhq/agent/TEMPLATES/Backstory.py

@@ -1,4 +1,4 @@
 BACKSTORY_FULL="""You are an expert {role} highly skilled in {skills}. You have abilities to query relevant information from the given knowledge sources and use tools such as {tools}. Leveraging these, you will identify competitive solutions to achieve the following goal: {goal}."""
 
 
-BACKSTORY_SHORT="""You are an expert {role} with relevant skillsets and abilities to query relevant information from the given knowledge sources. Leveraging these, you will identify competitive solutions to achieve the following goal: {goal}."""
+BACKSTORY_SHORT="""You are an expert {role} with relevant skills and abilities to query relevant information from the given knowledge sources. Leveraging these, you will identify competitive solutions to achieve the following goal: {goal}."""

versionhq/agent/model.py

@@ -32,13 +32,15 @@ class Agent(BaseModel):
     _request_within_rpm_limit: Any = PrivateAttr(default=None)
     _times_executed: int = PrivateAttr(default=0)
     _logger_config: Dict[str, Any] = PrivateAttr(default=dict(verbose=True, info_file_save=True))
+
+    api_key: Optional[str] = Field(default=None)
     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)
     role: str = Field(description="required. agent's role")
     goal: Optional[str] = Field(default=None)
     backstory: Optional[str] = Field(default=None, description="developer prompt to the llm")
-    skillsets: Optional[List[str]] = Field(default_factory=list)
+    skills: Optional[List[str]] = Field(default_factory=list, description="list up the agent's tangible skills in natural language")
     tools: Optional[List[Any]] = Field(default_factory=list)
 
     # knowledge
@@ -149,8 +151,12 @@ class Agent(BaseModel):
                                 setattr(tool, k, v)
                         tool_list.append(tool)
 
+                case callable():
+                    tool = Tool(func=item)
+                    tool_list.append(tool)
+
                 case _:
-                    if item.__base__ == BaseTool or item.__base__ == RagTool or item.__base__ == Tool:
+                    if hasattr(item, "__base__") and (item.__base__ == BaseTool or item.__base__ == RagTool or item.__base__ == Tool):
                         tool_list.append(item)
                     else:
                         Logger(**self._logger_config, filename=self.key).log(level="error", message=f"Tool {str(item)} is missing a function.", color="red")
@@ -169,12 +175,12 @@ class Agent(BaseModel):
         if self.backstory is None:
             from versionhq.agent.TEMPLATES.Backstory import BACKSTORY_FULL, BACKSTORY_SHORT
             backstory = ""
-            skills = ", ".join([item for item in self.skillsets]) if self.skillsets else ""
+            skills = ", ".join([item for item in self.skills]) if self.skills else ""
             tools = ", ".join([item.name for item in self.tools if hasattr(item, "name") and item.name is not None]) if self.tools else ""
             role = self.role.lower()
             goal = self.goal.lower() if self.goal else ""
 
-            if self.tools or self.skillsets:
+            if self.tools or self.skills:
                 backstory = BACKSTORY_FULL.format(role=role, goal=goal, skills=skills, tools=tools)
             else:
                 backstory = BACKSTORY_SHORT.format(role=role, goal=goal)
@@ -371,7 +377,7 @@ class Agent(BaseModel):
         messages = []
         messages.append({ "role": "user", "content": prompts })
         if self.use_developer_prompt:
-            messages.append({ "role": "system", "content": self.backstory })
+            messages.append({ "role": "developer", "content": self.backstory })
 
         try:
             if self._rpm_controller and self.max_rpm:

versionhq/llm/model.py

@@ -8,8 +8,7 @@ from contextlib import contextmanager
 from typing import Any, Dict, List, Optional
 from typing_extensions import Self
 
-import litellm
-from litellm import JSONSchemaValidationError
+from litellm import JSONSchemaValidationError, get_supported_openai_params
 from pydantic import BaseModel, Field, PrivateAttr, model_validator, ConfigDict
 
 from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES, MODELS, PARAMS, PROVIDERS, ENDPOINT_PROVIDERS, ENV_VARS
@@ -48,6 +47,7 @@ class FilteredStream:
 @contextmanager
 def suppress_warnings():
     with warnings.catch_warnings():
+        import litellm
         litellm.set_verbose = False
         warnings.filterwarnings(action="ignore")
         old_stdout = sys.stdout
@@ -94,8 +94,6 @@ class LLM(BaseModel):
 
     model_config = ConfigDict(extra="allow")
 
-    litellm.drop_params = True
-    litellm.set_verbose = True
     os.environ['LITELLM_LOG'] = 'DEBUG'
 
 
@@ -104,6 +102,8 @@ class LLM(BaseModel):
         """
         Validate the given model, provider, interface provider.
         """
+        import litellm
+        litellm.drop_params = True
 
         self._init_model_name = self.model
 
@@ -180,6 +180,9 @@ class LLM(BaseModel):
         """
         Set up valid config params after setting up a valid model, provider, interface provider names.
         """
+        import litellm
+        litellm.drop_params = True
+
         self._tokens = 0
 
         if self.callbacks:
@@ -202,11 +205,10 @@ class LLM(BaseModel):
         """
         Returns valid params incl. model + litellm original params) from the given config dict.
         """
-
         valid_config, valid_keys = dict(), list()
 
         if self.model:
-            valid_keys = litellm.get_supported_openai_params(
+            valid_keys = get_supported_openai_params(
                 model=self.model, custom_llm_provider=self.endpoint_provider, request_type="chat_completion"
             )
 
@@ -257,7 +259,7 @@ class LLM(BaseModel):
     def _supports_function_calling(self) -> bool:
         try:
             if self.model:
-                params = litellm.get_supported_openai_params(model=self.model)
+                params = get_supported_openai_params(model=self.model)
                 return "response_format" in params if params else False
             else:
                 return False
@@ -267,7 +269,8 @@ class LLM(BaseModel):
 
 
     def _supports_stop_words(self) -> bool:
-        supported_params = litellm.get_supported_openai_params(model=self.model, custom_llm_provider=self.endpoint_provider)
+        import litellm
+        supported_params = get_supported_openai_params(model=self.model, custom_llm_provider=self.endpoint_provider)
         return "stop" in supported_params if supported_params else False
 
 
@@ -279,6 +282,7 @@ class LLM(BaseModel):
 
 
     def _set_callbacks(self, callbacks: List[Any]):
+        import litellm
         callback_types = [type(callback) for callback in callbacks]
         for callback in litellm.success_callback[:]:
             if type(callback) in callback_types:
@@ -302,8 +306,9 @@ class LLM(BaseModel):
         """
         Execute LLM based on the agent's params and model params.
         """
-
+        import litellm
         litellm.drop_params = True
+        litellm.set_verbose = True
 
         with suppress_warnings():
             if len(self.callbacks) > 0:

{versionhq-1.2.3.4.dist-info → versionhq-1.2.3.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.3.4
+Version: 1.2.3.6
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -68,6 +68,9 @@ Requires-Dist: composio-core==0.7.0
 Requires-Dist: networkx>=3.4.2
 Requires-Dist: matplotlib>=3.10.0
 Requires-Dist: boto3>=1.37.1
+Requires-Dist: scikit-learn>=1.6.1
+Requires-Dist: numpy>=1.26.4
+Requires-Dist: pandas>=2.2.3
 Provides-Extra: docling
 Requires-Dist: docling>=2.25.2; extra == "docling"
 Provides-Extra: mem0ai
@@ -82,10 +85,6 @@ Requires-Dist: sec-api>=1.0.28; extra == "tools"
 Provides-Extra: torch
 Requires-Dist: torch>=2.6.0; extra == "torch"
 Requires-Dist: torchvision>=0.21.0; extra == "torch"
-Provides-Extra: evals
-Requires-Dist: scikit-learn>=1.6.1; extra == "evals"
-Requires-Dist: numpy>=1.26.4; extra == "evals"
-Requires-Dist: pandas>=2.2.3; extra == "evals"
 
 # Overview
 
@@ -143,9 +142,9 @@ Agentic orchestration framework for multi-agent networks and task graphs for com
 
 ## Key Features
 
-`versionhq` is a Python framework for agent networks that handle complex task automation without human interaction.
+`versionhq` is a Python framework designed for automating complex, multi-step tasks using autonomous agent networks.
 
-Agents are model-agnostic, and will improve task output, while optimizing token cost and job latency, by sharing their memory, knowledge base, and RAG tools with other agents in the network.
+Users can either configure their agents and network manually or allow the system to automatically manage the process based on provided task goals.
 
 
 ###  Agent Network
@@ -172,7 +171,7 @@ Each node is triggered by specific events and executed by an assigned agent once
 While the network automatically reconfigures itself, you retain the ability to direct the agents using `should_reform` variable.
 
 
-The following code snippet demonstrates the `TaskGraph` and its visualization, saving the diagram to the `uploads` directory.
+The following code snippet explicitly demonstrates the `TaskGraph` and its visualization, saving the diagram to the `uploads` directory.
 
 ```python
 import versionhq as vhq
@@ -214,15 +213,13 @@ A `TaskGraph` represents tasks as `nodes` and their execution dependencies as `e
 
 `Agent Networks` can handle `TaskGraph` objects by optimizing their formations.
 
-The following example demonstrates a simple concept of a `supervising` agent network handling a task graph with three tasks and one critical edge.
-
 <img src="https://res.cloudinary.com/dfeirxlea/image/upload/v1739337639/pj_m_home/zfg4ccw1m1ww1tpnb0pa.png">
 
 <hr />
 
 ### Optimization
 
-Agents are model-agnostic and can handle multiple tasks, leveraging their own and their peers' knowledge sources, memories, and tools.
+Autonomous agents are model-agnostic and can leverage their own and their peers' knowledge sources, memories, and tools.
 
 Agents are optimized during network formation, but customization is possible before or after.
 
@@ -581,7 +578,9 @@ Common issues and solutions:
 
 * Issues related to dependencies: `rm -rf uv.lock`, `uv cache clean`, `uv venv`, and run `uv pip install -r requirements.txt -v`.
 
-* Issues related to agents and other systems: Check `.logs` directory located in the root directory for error messages and stack traces.
+* Issues related to `torch` installation: Add optional dependencies by `uv add versionhq[torch]`.
+
+* Issues related to agents and other systems: Check `.logs` directory located at the root of the project directory for error messages and stack traces.
 
 * Issues related to `Python quit unexpectedly`: Check [this stackoverflow article](https://stackoverflow.com/questions/59888499/macos-catalina-python-quit-unexpectedly-error).
 

{versionhq-1.2.3.4.dist-info → versionhq-1.2.3.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=OB2B9WOc-P9RyhFLEhs2zS0E34d7GwED1POMAfSp4Cg,2980
+versionhq/__init__.py,sha256=40KCEmDQ4M8HPz4QLSdkRurG7jggekd83nKYJJ1j6dE,2980
 versionhq/_utils/__init__.py,sha256=d-vYVcORZKG-kkLe_fzE8VbViDpAk9DDOKe2fVK25ew,178
 versionhq/_utils/i18n.py,sha256=TwA_PnYfDLA6VqlUDPuybdV9lgi3Frh_ASsb_X8jJo8,1483
 versionhq/_utils/llm_as_a_judge.py,sha256=RM0oYfoeanuUyUL3Ewl6_8Xn1F5Axd285UMH46kxG1I,2378
@@ -8,10 +8,10 @@ versionhq/_utils/usage_metrics.py,sha256=xgYGRW3OTuK9EJyi3QYJeYcJl7dL27olcWaLo_7
 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=BPkvEyMH8VnZWsMeCwsGplDT_kLwlIejeRcr-6ItGqQ,2637
-versionhq/agent/model.py,sha256=JU5Yu2ODUAfODS5brm4yXsVWNGjjkJqfPcGJ1NZ8VnI,25392
+versionhq/agent/model.py,sha256=C3MptCdVF7c2xM5TR4Bz5TJzZw1MfYh50TLrvXFMrJ4,25656
 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/Backstory.py,sha256=dkfuATUQ2g2WoUKkmgAIch-RB--bektGoQaUlsDOn0g,529
 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=CYKNUeKC392wW3leIDIAaGiKADSsumC_vTe0VOnNwRs,7901
@@ -33,7 +33,7 @@ versionhq/knowledge/source_docling.py,sha256=dcu1ITqPXwWZ_lK-6tykEKhhC82eNRTMoWR
 versionhq/knowledge/storage.py,sha256=Kd-4r6aWM5EDaoXrzKXbgi1hY6tysSQARPGXM95qMmU,8266
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/llm/llm_vars.py,sha256=qSG-_pYeWksdMmwASXpjQqf97fMovsY4lNTSCHQF88k,5694
-versionhq/llm/model.py,sha256=R6Mlsfc_MHCPsaUhst1Q9K8KinjpVpKha-B8m6YRoIE,17230
+versionhq/llm/model.py,sha256=W9zcrC18zEz8BKKG2HJ2rO-8rdphif7sdiAHNc9oJbU,17400
 versionhq/memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/memory/contextual_memory.py,sha256=QEMVvHuEXxY7M6-12S8HhyFKf108KfX8Zzt7paPW048,3882
 versionhq/memory/model.py,sha256=VQR1229t7GQPMItlGAHLtJrb6LrZfSoRA1DRW4z0SOU,8234
@@ -62,8 +62,8 @@ versionhq/tool/decorator.py,sha256=C4ZM7Xi2gwtEMaSeRo-geo_g_MAkY77WkSLkAuY0AyI,1
 versionhq/tool/model.py,sha256=Nc2f9frTK5tH4kh6EeEAk1Fi1w19kEXLOcsBwHCS1a4,12189
 versionhq/tool/rag_tool.py,sha256=dW5o-83V4bMFFJEj3PUm7XjblwrYJGmZVBlCpPj6CeM,3852
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.2.3.4.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.3.4.dist-info/METADATA,sha256=6ctQ8pQeU6rOuUOO0AxAmVy4v0MYDk2QJbZyCWHFBR0,21418
-versionhq-1.2.3.4.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-versionhq-1.2.3.4.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.3.4.dist-info/RECORD,,
+versionhq-1.2.3.6.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.3.6.dist-info/METADATA,sha256=pse3tddq8a1S0jxvDrZBlb2v2a83PaF6oPRlSBvjXlM,21262
+versionhq-1.2.3.6.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+versionhq-1.2.3.6.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.3.6.dist-info/RECORD,,