versionhq 1.2.2.4__py3-none-any.whl → 1.2.2.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.2.4"
+__version__ = "1.2.2.6"
 __all__ = [
     "Agent",
 

versionhq/agent/inhouse_agents.py

@@ -38,7 +38,7 @@ vhq_formation_planner = Agent(
         "Solo is a formation where a single agent with tools, knowledge, and memory handles tasks indivudually. When self-learning mode is on - it will turn into Random formation. Typical usecase is an email agent drafts promo message for the given audience using their own knowledge.",
         "Supervising is a formation where the leader agent gives directions, while sharing its knowledge and memory with subbordinates.Subordinates can be solo agents or networks. Typical usecase is that the leader agent strategizes an outbound campaign plan and assigns components such as media mix or message creation to subordinate agents.",
         "Network is a formation where multple agents can share tasks, knowledge, and memory among network members without hierarchy. Typical usecase is that an email agent and social media agent share the product knowledge and deploy multi-channel outbound campaign. ",
-        "Random is a formation where a single agent handles tasks, asking help from other agents without sharing its memory or knowledge. Typical usecase is that an email agent drafts promo message for the given audience, asking insights on tones from other email agents which oversee other customer clusters, or an agent calls the external, third party agent to deploy the campaign. ",
+        "Random is a formation where a single agent handles tasks, asking help from other agents without sharing its memory or knowledge. Typical usecase is that an email agent drafts promo message for the given audience, asking insights on tones from other email agents which oversee other customer clusters, or an agent calls the external, third party agent to deploy the campaign.",
     ]
 )
 

versionhq/agent/model.py

@@ -35,8 +35,8 @@ class Agent(BaseModel):
     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="role of the agent - used in summary and logs")
-    goal: str = Field(description="concise goal of the agent (details are set in the Task instance)")
+    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)
     tools: Optional[List[Any]] = Field(default_factory=list)
@@ -68,10 +68,10 @@ class Agent(BaseModel):
     # llm settings cascaded to the LLM model
     llm: str | InstanceOf[LLM] | Dict[str, Any] = Field(default=None)
     func_calling_llm: str | InstanceOf[LLM] | Dict[str, Any] = Field(default=None)
-    respect_context_window: bool = Field(default=True,description="keep messages under the context window size")
+    respect_context_window: bool = Field(default=True, description="keep messages under the context window size")
     max_execution_time: Optional[int] = Field(default=None, description="max. task execution time in seconds")
     max_rpm: Optional[int] = Field(default=None, description="max. number of requests per minute")
-    llm_config: Optional[Dict[str, Any]] = Field(default=None, description="other llm config cascaded to the LLM model")
+    llm_config: Optional[Dict[str, Any]] = Field(default=None, description="other llm config cascaded to the LLM class")
 
     # # cache, error, ops handling
     # formatting_errors: int = Field(default=0, description="number of formatting errors.")
@@ -92,7 +92,7 @@ class Agent(BaseModel):
 
     @model_validator(mode="after")
     def validate_required_fields(self) -> Self:
-        required_fields = ["role", "goal"]
+        required_fields = ["role",]
         for field in required_fields:
             if getattr(self, field) is None:
                 raise ValueError(f"{field} must be provided either directly or through config")
@@ -172,7 +172,7 @@ class Agent(BaseModel):
             skills = ", ".join([item for item in self.skillsets]) if self.skillsets 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()
+            goal = self.goal.lower() if self.goal else ""
 
             if self.tools or self.skillsets:
                 backstory = BACKSTORY_FULL.format(role=role, goal=goal, skills=skills, tools=tools)
@@ -276,8 +276,8 @@ class Agent(BaseModel):
                 return self._set_llm_params(llm=llm, config=self.llm_config)
 
             case str():
-                llm_obj = LLM(model=llm)
-                return self._set_llm_params(llm=llm_obj, config=self.llm_config)
+                llm = LLM(model=llm)
+                return self._set_llm_params(llm=llm, config=self.llm_config)
 
             case dict():
                 model_name = llm.pop("model_name", llm.pop("deployment_name", str(llm)))
@@ -287,53 +287,21 @@ class Agent(BaseModel):
 
             case _:
                 model_name = (getattr(self.llm, "model_name") or getattr(self.llm, "deployment_name") or str(self.llm))
-                llm_obj = LLM(model=model_name if model_name else DEFAULT_MODEL_NAME)
+                llm = LLM(model=model_name if model_name else DEFAULT_MODEL_NAME)
                 llm_params = {
-                    "max_tokens": (getattr(llm, "max_tokens") or 3000),
                     "timeout": getattr(llm, "timeout", self.max_execution_time),
                     "callbacks": getattr(llm, "callbacks", None),
-                    "temperature": getattr(llm, "temperature", None),
-                    "logprobs": getattr(llm, "logprobs", None),
-                    "api_key": getattr(llm, "api_key", os.environ.get("LITELLM_API_KEY", None)),
+                    "llm_config": getattr(llm, "llm_config", None),
                     "base_url": getattr(llm, "base_url", None),
                 }
                 config = llm_params.update(self.llm_config) if self.llm_config else llm_params
-                return self._set_llm_params(llm=llm_obj, config=config)
+                return self._set_llm_params(llm=llm, config=config)
 
 
     def _set_llm_params(self, llm: LLM, config: Dict[str, Any] = None) -> LLM:
         """
         Add valid params to the LLM object.
         """
-
-        import litellm
-        from versionhq.llm.llm_vars import PARAMS
-
-        valid_config = {k: v for k, v in config.items() if v} if config else {}
-
-        if valid_config:
-            valid_keys = list()
-            try:
-                valid_keys = litellm.get_supported_openai_params(model=llm.model, custom_llm_provider=self.endpoint_provider, request_type="chat_completion")
-                if not valid_keys:
-                    valid_keys = PARAMS.get("common")
-            except:
-                valid_keys = PARAMS.get("common")
-
-            valid_keys += PARAMS.get("litellm")
-
-            for key in valid_keys:
-                if key in valid_config and valid_config[key]:
-                    val = valid_config[key]
-                    if [key == k for k, v in LLM.model_fields.items()]:
-                        setattr(llm, key, val)
-                    else:
-                        llm.other_valid_config.update({ key: val})
-
-
-        llm.timeout = self.max_execution_time if llm.timeout is None else llm.timeout
-        # llm.max_tokens = self.max_tokens if self.max_tokens else llm.max_tokens
-
         if llm.provider is None:
             provider_name = llm.model.split("/")[0]
             valid_provider = provider_name if provider_name in PROVIDERS else None
@@ -346,6 +314,12 @@ class Agent(BaseModel):
         if self.respect_context_window == False:
             llm.context_window_size = DEFAULT_CONTEXT_WINDOW_SIZE
 
+        llm.timeout = self.max_execution_time if llm.timeout is None else llm.timeout
+
+        if config:
+            llm.llm_config = {k: v for k, v in config.items() if v or v == False}
+            llm.setup_config()
+
         return llm
 
 
@@ -494,7 +468,7 @@ class Agent(BaseModel):
         Defines and executes a task when it is not given and returns TaskOutput object.
         """
 
-        if not self.goal or not self.role:
+        if not self.role:
             return None
 
         from versionhq.task.model import Task
@@ -504,7 +478,7 @@ class Agent(BaseModel):
             steps: list[str]
 
         task = Task(
-            description=f"Generate a simple result in a sentence to achieve the goal: {self.goal}. If needed, list up necessary steps in concise manner.",
+            description=f"Generate a simple result in a sentence to achieve the goal: {self.goal if self.goal else self.role}. If needed, list up necessary steps in concise manner.",
             pydantic_output=Output,
             tool_res_as_final=tool_res_as_final,
         )
@@ -595,7 +569,7 @@ class Agent(BaseModel):
 
 
     def __repr__(self):
-        return f"Agent(role={self.role}, goal={self.goal}"
+        return f"Agent(role={self.role}, id={str(self.id)}"
 
     def __str__(self):
         return super().__str__()

versionhq/llm/llm_vars.py

@@ -6,30 +6,16 @@ PROVIDERS = [
     "openai",
     "gemini",
     "openrouter",
-    "huggingface",
     "anthropic",
-    "sagemaker",
     "bedrock",
-    "ollama",
-    "watson",
-    "azure",
-    "cerebras",
-    "llama",
+    "bedrock/converse",
+    "huggingface",
 ]
 
 ENDPOINT_PROVIDERS = [
     "huggingface",
 ]
 
-"""
-List of models available on the framework.
-Model names align with the LiteLLM's key names defined in the JSON URL.
-Provider names align with the custom provider or model provider names.
--> model_key = custom_provider_name/model_name
-
-Option
-litellm.pick_cheapest_chat_models_from_llm_provider(custom_llm_provider: str, n=1)
-"""
 
 MODELS = {
     "openai": [
@@ -45,6 +31,7 @@ MODELS = {
         "gemini/gemini-2.0-flash-exp",
     ],
     "anthropic": [
+        "claude-3-7-sonnet-latest",
         "claude-3-5-sonnet-20241022",
         "claude-3-5-sonnet-20240620",
         "claude-3-haiku-2024030",
@@ -53,77 +40,52 @@ MODELS = {
     ],
     "openrouter": [
         "openrouter/deepseek/deepseek-r1",
+
         "openrouter/qwen/qwen-2.5-72b-instruct",
+
         "openrouter/google/gemini-2.0-flash-thinking-exp:free",
         "openrouter/google/gemini-2.0-flash-thinking-exp-1219:free",
         "openrouter/google/gemini-2.0-flash-001",
+
         "openrouter/meta-llama/llama-3.3-70b-instruct",
         "openrouter/mistralai/mistral-large-2411",
+        "openrouter/cohere/command-r-plus",
     ],
-    "huggingface": [
-        "huggingface/qwen/qwen2.5-VL-72B-Instruct",
-    ],
-    # "sagemaker": [
-    #     "sagemaker/huggingface-text2text-flan-t5-base",
-    #     "sagemaker/huggingface-llm-gemma-7b",
-    #     "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-13b",
-    #     "sagemaker/jumpstart-dft-meta-textgeneration-llama-2-70b",
-    #     "sagemaker/jumpstart-dft-meta-textgeneration-llama-3-8b",
-    #     "sagemaker/jumpstart-dft-meta-textgeneration-llama-3-70b",
-    #     "sagemaker/huggingface-llm-mistral-7b"
-    # ], #https://docs.aws.amazon.com/sagemaker/latest/dg/jumpstart-foundation-models-latest.html
-    "ollama": [
-      "ollama/llama3.1",
-      "ollama/mixtral",
-      "ollama/mixtral-8x22B-Instruct-v0.1",
-    ],
-    # "watson": [
-    #     "watsonx/meta-llama/llama-3-1-70b-instruct",
-    #     "watsonx/meta-llama/llama-3-1-8b-instruct",
-    #     "watsonx/meta-llama/llama-3-2-11b-vision-instruct",
-    #     "watsonx/meta-llama/llama-3-2-1b-instruct",
-    #     "watsonx/meta-llama/llama-3-2-90b-vision-instruct",
-    #     "watsonx/meta-llama/llama-3-405b-instruct",
-    #     "watsonx/mistral/mistral-large",
-    #     "watsonx/ibm/granite-3-8b-instruct",
-    # ],
     "bedrock": [
-        "bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0",
-        "bedrock/anthropic.claude-3-sonnet-20240229-v1:0",
-        "bedrock/anthropic.claude-3-haiku-20240307-v1:0",
-        "bedrock/anthropic.claude-3-opus-20240229-v1:0",
-        "bedrock/anthropic.claude-v2",
-        "bedrock/anthropic.claude-instant-v1",
-        "bedrock/meta.llama3-1-405b-instruct-v1:0",
-        "bedrock/meta.llama3-1-70b-instruct-v1:0",
-        "bedrock/meta.llama3-1-8b-instruct-v1:0",
-        "bedrock/meta.llama3-70b-instruct-v1:0",
-        "bedrock/meta.llama3-8b-instruct-v1:0",
+        "bedrock/converse/us.meta.llama3-3-70b-instruct-v1:0",
+        "bedrock/us.meta.llama3-2-1b-instruct-v1:0",
+        "bedrock/us.meta.llama3-2-3b-instruct-v1:0",
+        "bedrock/us.meta.llama3-2-11b-instruct-v1:0",
+
+        "bedrock/mistral.mistral-7b-instruct-v0:2",
+        "bedrock/mistral.mixtral-8x7b-instruct-v0:1",
+        "bedrock/mistral.mistral-large-2407-v1:0",
+
         "bedrock/amazon.titan-text-lite-v1",
         "bedrock/amazon.titan-text-express-v1",
+        "bedrock/amazon.titan-text-premier-v1:0",
+
+        "bedrock/cohere.command-r-plus-v1:0",
+        "bedrock/cohere.command-r-v1:0",
         "bedrock/cohere.command-text-v14",
-        "bedrock/ai21.j2-mid-v1",
-        "bedrock/ai21.j2-ultra-v1",
-        "bedrock/ai21.jamba-instruct-v1:0",
-        "bedrock/meta.llama2-13b-chat-v1",
-        "bedrock/meta.llama2-70b-chat-v1",
-        "bedrock/mistral.mistral-7b-instruct-v0:2",
-        "bedrock/mistral.mixtral-8x7b-instruct-v0:1",
+        "bedrock/cohere.command-light-text-v14",
+    ],
+    "huggingface": [
+        "huggingface/qwen/qwen2.5-VL-72B-Instruct",
     ],
 }
 
 
-
-KEYS = {
+ENV_VARS = {
     "openai": ["OPENAI_API_KEY"],
     "gemini": ["GEMINI_API_KEY"],
     "anthropic": ["ANTHROPIC_API_KEY"],
     "huggingface": ["HUGGINGFACE_API_KEY", ],
-    "sagemaker": ["AWS_ACCESS_KEY_ID", "ADW_SECURET_ACCESS_KEY", "AWS_REGION_NAME"],
+    "bedrock": ["AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_REGION_NAME"],
+    "sagemaker": ["AWS_ACCESS_KEY_ID", "AWS_SECRET_ACCESS_KEY", "AWS_REGION_NAME"],
 }
 
 
-
 """
 Max input token size by the model.
 """
@@ -193,8 +155,8 @@ PARAMS = {
         "response_format",
         "n",
         "stop",
-        "base_url",
-        "api_key",
+        # "base_url",
+        # "api_key",
     ],
     "openai": [
         "timeout",
@@ -216,7 +178,10 @@ PARAMS = {
     ],
     "gemini": [
         "topK",
-    ]
+    ],
+    "bedrock": {
+        "top-k",
+    }
 }
 
 

versionhq/llm/model.py

@@ -4,14 +4,15 @@ import sys
 import threading
 import warnings
 from dotenv import load_dotenv
-import litellm
-from litellm import JSONSchemaValidationError
 from contextlib import contextmanager
 from typing import Any, Dict, List, Optional
 from typing_extensions import Self
+
+import litellm
+from litellm import JSONSchemaValidationError
 from pydantic import BaseModel, Field, PrivateAttr, model_validator, ConfigDict
 
-from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES, MODELS, PARAMS, PROVIDERS, ENDPOINT_PROVIDERS
+from versionhq.llm.llm_vars import LLM_CONTEXT_WINDOW_SIZES, MODELS, PARAMS, PROVIDERS, ENDPOINT_PROVIDERS, ENV_VARS
 from versionhq.tool.model import Tool, ToolSet
 from versionhq._utils.logger import Logger
 
@@ -68,36 +69,23 @@ class LLM(BaseModel):
 
     _logger: Logger = PrivateAttr(default_factory=lambda: Logger(verbose=True))
     _init_model_name: str = PrivateAttr(default=None)
-    _tokens: int = PrivateAttr(default=0) # accumulate total tokens used for the call
-    model_config = ConfigDict(extra="allow")
+    # _init_config: Optional[Dict[str, Any]] = PrivateAttr(default_factory=dict) # stores llm config passed by client or agent
+    _tokens: int = PrivateAttr(default=0) # aggregate number of tokens consumed
 
     model: str = Field(default=None)
     provider: Optional[str] = Field(default=None, description="model provider")
-    endpoint_provider: Optional[str] = Field(default=None, description="custom endpoint provider for pass through llm call. must need base_url")
+    endpoint_provider: Optional[str] = Field(default=None, description="custom endpoint provider for pass through llm call. require base_url")
     base_url: Optional[str] = Field(default=None, description="api base url for endpoint provider")
-    api_key: Optional[str] = Field(default=None, description="api key to access the model")
 
     # optional params
     response_format: Optional[Any] = Field(default=None)
+    llm_config: Optional[Dict[str, Any]] = Field(default_factory=dict, description="stores valid llm config params")
+    callbacks: Optional[List[Any]] = Field(default_factory=list)
+    tools: Optional[List[Dict[str, Any]]] = Field(default_factory=list, description="stores a list of tool properties")
     timeout: Optional[float | int] = Field(default=None)
-    max_tokens: Optional[int] = Field(default=None)
-    max_completion_tokens: Optional[int] = Field(default=None)
     context_window_size: Optional[int] = Field(default=DEFAULT_CONTEXT_WINDOW_SIZE)
-    temperature: Optional[float] = Field(default=None)
-    top_p: Optional[float] = Field(default=None)
-    n: Optional[int] = Field(default=None)
-    stop: Optional[str | List[str]] = Field(default=None)
-    presence_penalty: Optional[float] = Field(default=None)
-    frequency_penalty: Optional[float] = Field(default=None)
-    logit_bias: Optional[Dict[int, float]] = Field(default=None)
-    seed: Optional[int] = Field(default=None)
-    logprobs: Optional[bool] = Field(default=None)
-    top_logprobs: Optional[int] = Field(default=None)
-    tools: Optional[List[Dict[str, Any]]] = Field(default_factory=list, description="store a list of tool properties")
-    callbacks: List[Any] = Field(default_factory=list)
-    other_valid_config: Optional[Dict[str, Any]] = Field(default_factory=dict, description="store other valid values in dict to cascade to the model")
-
-    # LiteLLM specific fields
+
+    # LiteLLM specific config
     api_base: Optional[str] = Field(default=None, description="litellm specific field - api base of the model provider")
     api_version: Optional[str] = Field(default=None)
     num_retries: Optional[int] = Field(default=1)
@@ -105,6 +93,8 @@ class LLM(BaseModel):
     fallbacks: Optional[List[Any]]= Field(default=None, description="A list of model names + params to be used, in case the initial call fails")
     metadata: Optional[Dict[str, Any]] = Field(default=None)
 
+    model_config = ConfigDict(extra="allow")
+
     litellm.drop_params = True
     litellm.set_verbose = True
     os.environ['LITELLM_LOG'] = 'DEBUG'
@@ -187,9 +177,9 @@ class LLM(BaseModel):
 
 
     @model_validator(mode="after")
-    def validate_model_params(self) -> Self:
+    def setup_config(self) -> Self:
         """
-        Set up valid params to the model after setting up a valid model, provider, interface provider names.
+        Set up valid config params after setting up a valid model, provider, interface provider names.
         """
         self._tokens = 0
 
@@ -198,28 +188,28 @@ class LLM(BaseModel):
 
         self.context_window_size = self._get_context_window_size()
 
-        api_key_name = self.provider.upper() + "_API_KEY" if self.provider else None
-        if api_key_name:
-            self.api_key = os.environ.get(api_key_name, None)
-
         base_url_key_name = self.endpoint_provider.upper() + "_API_BASE" if self.endpoint_provider else None
-
         if base_url_key_name:
             self.base_url = os.environ.get(base_url_key_name)
             self.api_base = self.base_url
 
+        if self.llm_config:
+            self._create_valid_params(config=self.llm_config)
+
         return self
 
 
     def _create_valid_params(self, config: Dict[str, Any]) -> Dict[str, Any]:
         """
-        Return valid params (model + litellm original params) from the given config dict.
+        Returns valid params incl. model + litellm original params) from the given config dict.
         """
 
-        valid_params, valid_keys = dict(), list()
+        valid_config, valid_keys = dict(), list()
 
         if self.model:
-            valid_keys = litellm.get_supported_openai_params(model=self.model, custom_llm_provider=self.endpoint_provider, request_type="chat_completion")
+            valid_keys = litellm.get_supported_openai_params(
+                model=self.model, custom_llm_provider=self.endpoint_provider, request_type="chat_completion"
+            )
 
         if not valid_keys:
             valid_keys = PARAMS.get("common")
@@ -227,14 +217,38 @@ class LLM(BaseModel):
         valid_keys += PARAMS.get("litellm")
 
         for item in valid_keys:
-            if hasattr(self, item) and getattr(self, item):
-                valid_params[item] = getattr(self, item)
-            elif item in self.other_valid_config and self.other_valid_config[item]:
-                valid_params[item] = self.other_valid_config[item]
-            elif item in config and config[item]:
-                valid_params[item] = config[item]
+            if hasattr(self, item) and (getattr(self, item) or getattr(self, item) == False):
+                valid_config[item] = getattr(self, item)
 
-        return valid_params
+            elif item in self.llm_config and (self.llm_config[item] or self.llm_config[item]==False):
+                valid_config[item] = self.llm_config[item]
+
+            elif item in config and (config[item] or config[item] == False):
+                valid_config[item] = config[item]
+
+            else:
+                pass
+
+        self.llm_config = valid_config
+        return valid_config
+
+
+    def _set_env_vars(self) -> Dict[str, Any]:
+        if self.provider == "openai":
+            return {}
+
+        cred = dict()
+        env_vars = ENV_VARS.get(self.provider, None) if self.provider else None
+
+        if not env_vars:
+            return {}
+
+        for item in env_vars:
+            val = os.environ.get(item, None)
+            if val:
+                cred[str(item).lower()] = val
+
+        return cred
 
 
     def _supports_function_calling(self) -> bool:
@@ -242,6 +256,8 @@ class LLM(BaseModel):
             if self.model:
                 params = litellm.get_supported_openai_params(model=self.model)
                 return "response_format" in params if params else False
+            else:
+                return False
         except Exception as e:
             self._logger.log(level="warning", message=f"Failed to get supported params: {str(e)}", color="yellow")
             return False
@@ -288,15 +304,16 @@ class LLM(BaseModel):
 
         with suppress_warnings():
             if len(self.callbacks) > 0:
-                self._set_callbacks(self.callbacks) # passed by agent
+                self._set_callbacks(self.callbacks)
 
             try:
                 res, tool_res = None, ""
+                cred = self._set_env_vars()
 
                 if not tools:
                     self.response_format = response_format
                     params = self._create_valid_params(config=config)
-                    res = litellm.completion(model=self.model, messages=messages, stream=False, **params)
+                    res = litellm.completion(model=self.model, messages=messages, stream=False, **params, **cred)
                     self._tokens += int(res["usage"]["total_tokens"])
                     return res["choices"][0]["message"]["content"]
 
@@ -305,7 +322,7 @@ class LLM(BaseModel):
                         self.response_format = { "type": "json_object" }  if tool_res_as_final and self.provider != "gemini" else response_format
                         self.tools = [item.tool.properties if isinstance(item, ToolSet) else item.properties for item in tools]
                         params = self._create_valid_params(config=config)
-                        res = litellm.completion(model=self.model, messages=messages, **params)
+                        res = litellm.completion(model=self.model, messages=messages, **params, **cred)
                         tool_calls = res.choices[0].message.tool_calls
 
                         if tool_calls:
@@ -367,7 +384,7 @@ class LLM(BaseModel):
                 if tool_res_as_final:
                     return tool_res
                 else:
-                    res = litellm.completion(model=self.model, messages=messages, **params)
+                    res = litellm.completion(model=self.model, messages=messages, **params, **cred)
                     self._tokens += int(res["usage"]["total_tokens"])
                     return res.choices[0].message.content
 

versionhq/task/model.py

@@ -709,11 +709,11 @@ Ref. Output image: {output_formats_to_follow}
         #     )
         #     self._save_file(content)
 
-
-        # successful output will be evaluated and stored in the logs
         if raw_output:
             if self.should_evaluate:
                 task_output.evaluate(task=self)
+                self.output = task_output
+
             self._create_short_and_long_term_memories(agent=agent, task_output=task_output)
 
             if self.callback and isinstance(self.callback, Callable):
@@ -723,7 +723,7 @@ Ref. Output image: {output_formats_to_follow}
                 valid_kwargs = { k: kwargs[k] if  k in kwargs else None for k in valid_keys }
                 callback_res = self.callback(**valid_kwargs)
                 task_output.callback_output = callback_res
-
+                self.output = task_output
             self._store_logs()
 
         return task_output

{versionhq-1.2.2.4.dist-info → versionhq-1.2.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.2.4
+Version: 1.2.2.6
 Summary: An agentic orchestration framework for building agent networks that handle task automation.
 Author-email: Kuriko Iwai <kuriko@versi0n.io>
 License: MIT License
@@ -66,6 +66,7 @@ Requires-Dist: envoy>=0.0.3
 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
 Provides-Extra: docling
 Requires-Dist: docling>=2.17.0; extra == "docling"
 Provides-Extra: mem0ai
@@ -141,7 +142,7 @@ Agentic orchestration framework for multi-agent networks and task graphs for com
 
 `versionhq` is a Python framework for agent networks that handle complex task automation without human interaction.
 
-Agents are model-agnostic, and will improve task output, while oprimizing token cost and job latency, by sharing their memory, knowledge base, and RAG tools with other agents in the network.
+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.
 
 
 ###  Agent Network
@@ -227,10 +228,7 @@ The following code snippet demonstrates agent customization:
 ```python
 import versionhq as vhq
 
-agent = vhq.Agent(
-   role="Marketing Analyst",
-   goal="my amazing goal"
-) # assuming this agent was created during the network formation
+agent = vhq.Agent(role="Marketing Analyst")
 
 # update the agent
 agent.update(
@@ -321,8 +319,8 @@ To create an agent network with one or more manager agents, designate members us
 ```python
 import versionhq as vhq
 
-agent_a = vhq.Agent(role="agent a", goal="My amazing goals", llm="llm-of-your-choice")
-agent_b = vhq.Agent(role="agent b", goal="My amazing goals", llm="llm-of-your-choice")
+agent_a = vhq.Agent(role="agent a", llm="llm-of-your-choice")
+agent_b = vhq.Agent(role="agent b", llm="llm-of-your-choice")
 
 task_1 = vhq.Task(
    description="Analyze the client's business model.",

{versionhq-1.2.2.4.dist-info → versionhq-1.2.2.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-versionhq/__init__.py,sha256=PqOgmog9JVNxGm2LP1RV5x3D-U7PClfcKFWVAE6dXCM,2980
+versionhq/__init__.py,sha256=LPGLlfGPZ6saKcuLrozmgUNP89Po4H5uML4Iw62OjkE,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
@@ -7,8 +7,8 @@ versionhq/_utils/process_config.py,sha256=YTGY_erW335RfceQfzS18YAqq-AAb-iSvKSjN7
 versionhq/_utils/usage_metrics.py,sha256=xgYGRW3OTuK9EJyi3QYJeYcJl7dL27olcWaLo_7B3JE,2246
 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=WAbyog-6pKwa8ru9u_KJgD_ViTLv4ZRECks1Znch47E,2638
-versionhq/agent/model.py,sha256=xebClhCdx0Xt-_or3Ne_fTh8eOWSDEbi4r3WvGSOYcI,26595
+versionhq/agent/inhouse_agents.py,sha256=BPkvEyMH8VnZWsMeCwsGplDT_kLwlIejeRcr-6ItGqQ,2637
+versionhq/agent/model.py,sha256=JU5Yu2ODUAfODS5brm4yXsVWNGjjkJqfPcGJ1NZ8VnI,25392
 versionhq/agent/parser.py,sha256=riG0dkdQCxH7uJ0AbdVdg7WvL0BXhUgJht0VtQvxJBc,4082
 versionhq/agent/rpm_controller.py,sha256=grezIxyBci_lDlwAlgWFRyR5KOocXeOhYkgN02dNFNE,2360
 versionhq/agent/TEMPLATES/Backstory.py,sha256=IAhGnnt6VUMe3wO6IzeyZPDNu7XE7Uiu3VEXUreOcKs,532
@@ -32,8 +32,8 @@ versionhq/knowledge/source.py,sha256=-hEUPtJUHHMx4rUKtiHl19J8xAMw-WVBw34zwa2jZ08
 versionhq/knowledge/source_docling.py,sha256=dcu1ITqPXwWZ_lK-6tykEKhhC82eNRTMoWRpxK9Kzls,5441
 versionhq/knowledge/storage.py,sha256=Kd-4r6aWM5EDaoXrzKXbgi1hY6tysSQARPGXM95qMmU,8266
 versionhq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-versionhq/llm/llm_vars.py,sha256=wjQK20cKvph6Vq1v71o4d16zBGcHlwq0bzOT_zWno7w,7041
-versionhq/llm/model.py,sha256=HIBmf8FYV6-cDbZK1ZBu6z3dmF0ZUbKbCelfwxMlgyY,17177
+versionhq/llm/llm_vars.py,sha256=nqnSyuPgIwXnL4JVgC0DFkAe-mV2n3eUd7mtFkfGfds,5426
+versionhq/llm/model.py,sha256=PZm4gvQNJTW_ZwJE4MEF9JC05zbxS8RfSFQ8pFcNW-A,17146
 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
@@ -47,7 +47,7 @@ versionhq/storage/utils.py,sha256=r5ghA_ktdR2IuzlzKqZYCjsNxztEMzyhWLneA4cFuWY,74
 versionhq/task/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 versionhq/task/evaluation.py,sha256=iRLzppqwKaiGpbsr9gMbf6T7NQe6rxTA6OBcWhmiCKs,4473
 versionhq/task/formatter.py,sha256=N8Kmk9vtrMtBdgJ8J7RmlKNMdZWSmV8O1bDexmCWgU0,643
-versionhq/task/model.py,sha256=8qBxRgqMs_b8IZA45_gAU1uc5S5w9hdnyRG-Vwm8bVM,28901
+versionhq/task/model.py,sha256=rJmyEUM1DnX1mkN_0etYXqBJP_YADXVXZhFR5R-h8ZA,28915
 versionhq/task/structured_response.py,sha256=4q-hQPu7oMMHHXEzh9YW4SJ7N5eCZ7OfZ65juyl_jCI,5000
 versionhq/task/TEMPLATES/Description.py,sha256=EkwJHc65G32MjWyn3rcp0ATmMaVPHuYKaykyByU5r4g,751
 versionhq/task_graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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=qm_nDWs-WyDvrxZeZAL2AkswfUWGPZS4zybz0o6wOFI,3653
 versionhq/tool/tool_handler.py,sha256=2m41K8qo5bGCCbwMFferEjT-XZ-mE9F0mDUOBkgivOI,1416
-versionhq-1.2.2.4.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
-versionhq-1.2.2.4.dist-info/METADATA,sha256=fX-azucXgCg-Fm05snnLT02ttDz5u10xMOaBjGmpkp4,22213
-versionhq-1.2.2.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-versionhq-1.2.2.4.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
-versionhq-1.2.2.4.dist-info/RECORD,,
+versionhq-1.2.2.6.dist-info/LICENSE,sha256=cRoGGdM73IiDs6nDWKqPlgSv7aR4n-qBXYnJlCMHCeE,1082
+versionhq-1.2.2.6.dist-info/METADATA,sha256=OwLIKXnR7HIZiJc1ZeuK9zu_65eXSMgObNOzuAg0o4c,22097
+versionhq-1.2.2.6.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+versionhq-1.2.2.6.dist-info/top_level.txt,sha256=DClQwxDWqIUGeRJkA8vBlgeNsYZs4_nJWMonzFt5Wj0,10
+versionhq-1.2.2.6.dist-info/RECORD,,

{versionhq-1.2.2.4.dist-info → versionhq-1.2.2.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (75.8.2)
 Root-Is-Purelib: true
 Tag: py3-none-any