versionhq 1.2.2.5__tar.gz → 1.2.2.7__tar.gz

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/.env.sample

@@ -16,3 +16,7 @@ COMPOSIO_API_KEY=
 COMPOSIO_CLI_KEY=
 
 MEM0_API_KEY=
+
+AWS_ACCESS_KEY_ID=
+AWS_SECRET_ACCESS_KEY=
+AWS_REGION_NAME=

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/.github/workflows/run_tests.yml

@@ -17,6 +17,9 @@ env:
   DEFAULT_REDIRECT_URL:  ${{ secrets.DEFAULT_REDIRECT_URL }}
   DEFAULT_USER_ID:  ${{ secrets.DEFAULT_USER_ID }}
   MEM0_API_KEY: ${{ secrets.MEM0_API_KEY }}
+  AWS_ACCESS_KEY_ID: ${{ secrets.AWS_ACCESS_KEY_ID }}
+  AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_SECRET_ACCESS_KEY }}
+  AWS_REGION_NAME: ${{ secrets.AWS_REGION_NAME }}
 
 jobs:
   run_test:

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: versionhq
-Version: 1.2.2.5
+Version: 1.2.2.7
 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
@@ -98,10 +99,10 @@ Agentic orchestration framework for multi-agent networks and task graphs for com
 
 **Visit:**
 
-- [Playground](https://versi0n.io/playground)
+- [Playground](https://versi0n.io/)
 - [Docs](https://docs.versi0n.io)
-- [Github Repository](https://github.com/versionHQ/multi-agent-system)
-- [PyPI](https://pypi.org/project/versionhq/)
+- [Github](https://github.com/versionHQ/)
+- [Python SDK](https://pypi.org/project/versionhq/)
 
 <hr />
 
@@ -116,9 +117,10 @@ Agentic orchestration framework for multi-agent networks and task graphs for com
   - [Optimization](#optimization)
 - [Quick Start](#quick-start)
   - [Package installation](#package-installation)
-  - [Forming a agent network](#forming-a-agent-network)
-  - [Executing tasks](#executing-tasks)
-  - [Supervising](#supervising)
+  - [Launching an agent](#launching-an-agent)
+  - [Automating workflows](#automating-workflows)
+  - [Executing a single task](#executing-a-single-task)
+  - [Supervising agents](#supervising-agents)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
 - [Setting Up Your Project](#setting-up-your-project)
@@ -141,7 +143,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
@@ -246,33 +248,45 @@ agent.update(
 
 ### Package installation
 
-   ```
-   pip install versionhq
-   ```
+```
+pip install versionhq
+```
 
 (Python 3.11 / 3.12)
 
-### Forming a agent network
 
-   ```python
-   import versionhq as vhq
+### Launching an agent
 
-   network = work(
-      task="YOUR AMAZING TASK OVERVIEW",
-      expected_outcome="YOUR OUTCOME EXPECTATION",
-   )
-   res, tg = network.launch()
-   ```
 
-This will form a agent network with multiple agents on `Formation` and return response in `TaskOutput` object and `TaskGraph` that connects multiple tasks as nodes.
+```python
+import versionhq as vhq
 
+agent = vhq.Agent(role="Marketer")
+res = agent.start()
+
+assert isinstance(res, vhq.TaskOutput) # contains agent's response in text, JSON, Pydantic formats with usage recordes and eval scores.
+```
 
-### Executing tasks
 
-You can simply build an agent using `Agent` model and execute the task using `Task` class.
+### Automating workflows
 
-By default, agents prioritize JSON over plane text outputs.
+```python
+import versionhq as vhq
 
+network = vhq.form_agent_network(
+   task="draft a promo plan",
+   expected_outcome="marketing plan, budget, KPI targets",
+)
+res, tg = network.launch()
+
+assert isinstance(res, vhq.TaskOutput) # the latest output from the workflow
+assert isinstance(tg, vhq.TaskGraph) # contains task nodes and edges that connect the nodes with dep-met conditions
+```
+
+
+### Executing a single task
+
+You can simply build and execute a task using `Task` class.
 
 ```python
 import versionhq as vhq
@@ -292,34 +306,20 @@ task = vhq.Task(
    callback_kwargs=dict(message="Hi! Here is the result: ")
 )
 
-res = task.execute(context="context to consider")
-
+res = task.execute(context="testing a task function")
 assert isinstance(res, vhq.TaskOutput)
 ```
 
-This will return a `TaskOutput` object that stores response in plane text, JSON, and Pydantic model: `CustomOutput` formats with a callback result, tool output (if given), and evaluation results (if given).
 
-```python
-res == vhq.TaskOutput(
-   task_id=UUID('<TASK UUID>'),
-   raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
-   json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
-   pydantic=<class '__main__.CustomOutput'>,
-   tool_output=None,
-   callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
-   evaluation=None
-)
-```
-
-### Supervising
+### Supervising agents
 
 To create an agent network with one or more manager agents, designate members using the `is_manager` tag.
 
 ```python
 import versionhq as vhq
 
-agent_a = vhq.Agent(role="agent a", llm="llm-of-your-choice")
-agent_b = vhq.Agent(role="agent b", llm="llm-of-your-choice")
+agent_a = vhq.Agent(role="Member", llm="gpt-4o")
+agent_b = vhq.Agent(role="Leader", llm="gemini-2.0")
 
 task_1 = vhq.Task(
    description="Analyze the client's business model.",
@@ -341,11 +341,9 @@ network =vhq.AgentNetwork(
 )
 res, tg = network.launch()
 
-assert isinstance(res, vhq.TaskOutput)
-assert agent_b.key in task_1.processed_agents
-assert agent_b.key in task_2.processed_agents
-
-assert isinstance(tg, vhq.TaskGraph)
+assert isinstance(res, vhq.NetworkOutput)
+assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
+assert [item for item in task_1.processed_agents if "agent b" == item]
 ```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.
@@ -509,7 +507,7 @@ Create `.env` file in the project root and add secret vars following `.env.sampl
 
    * Test functions within the files must begin with `test_`.
 
-   * Pytest priorities are `1. playground demo > 2. docs use cases > 3. other features`
+   * Pytest priorities are `1. playground > 2. docs use cases > 3. other features`
 
 
 4. Update `docs` accordingly.
@@ -597,4 +595,4 @@ Common issues and solutions:
 ## Frequently Asked Questions (FAQ)
 **Q. Where can I see if the agent is working?**
 
-A. Visit [playground](https://versi0n.io/playground).
+A. Visit [playground](https://versi0n.io).

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/README.md

@@ -12,10 +12,10 @@ Agentic orchestration framework for multi-agent networks and task graphs for com
 
 **Visit:**
 
-- [Playground](https://versi0n.io/playground)
+- [Playground](https://versi0n.io/)
 - [Docs](https://docs.versi0n.io)
-- [Github Repository](https://github.com/versionHQ/multi-agent-system)
-- [PyPI](https://pypi.org/project/versionhq/)
+- [Github](https://github.com/versionHQ/)
+- [Python SDK](https://pypi.org/project/versionhq/)
 
 <hr />
 
@@ -30,9 +30,10 @@ Agentic orchestration framework for multi-agent networks and task graphs for com
   - [Optimization](#optimization)
 - [Quick Start](#quick-start)
   - [Package installation](#package-installation)
-  - [Forming a agent network](#forming-a-agent-network)
-  - [Executing tasks](#executing-tasks)
-  - [Supervising](#supervising)
+  - [Launching an agent](#launching-an-agent)
+  - [Automating workflows](#automating-workflows)
+  - [Executing a single task](#executing-a-single-task)
+  - [Supervising agents](#supervising-agents)
 - [Technologies Used](#technologies-used)
 - [Project Structure](#project-structure)
 - [Setting Up Your Project](#setting-up-your-project)
@@ -55,7 +56,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
@@ -160,33 +161,45 @@ agent.update(
 
 ### Package installation
 
-   ```
-   pip install versionhq
-   ```
+```
+pip install versionhq
+```
 
 (Python 3.11 / 3.12)
 
-### Forming a agent network
 
-   ```python
-   import versionhq as vhq
+### Launching an agent
 
-   network = work(
-      task="YOUR AMAZING TASK OVERVIEW",
-      expected_outcome="YOUR OUTCOME EXPECTATION",
-   )
-   res, tg = network.launch()
-   ```
 
-This will form a agent network with multiple agents on `Formation` and return response in `TaskOutput` object and `TaskGraph` that connects multiple tasks as nodes.
+```python
+import versionhq as vhq
 
+agent = vhq.Agent(role="Marketer")
+res = agent.start()
+
+assert isinstance(res, vhq.TaskOutput) # contains agent's response in text, JSON, Pydantic formats with usage recordes and eval scores.
+```
 
-### Executing tasks
 
-You can simply build an agent using `Agent` model and execute the task using `Task` class.
+### Automating workflows
 
-By default, agents prioritize JSON over plane text outputs.
+```python
+import versionhq as vhq
 
+network = vhq.form_agent_network(
+   task="draft a promo plan",
+   expected_outcome="marketing plan, budget, KPI targets",
+)
+res, tg = network.launch()
+
+assert isinstance(res, vhq.TaskOutput) # the latest output from the workflow
+assert isinstance(tg, vhq.TaskGraph) # contains task nodes and edges that connect the nodes with dep-met conditions
+```
+
+
+### Executing a single task
+
+You can simply build and execute a task using `Task` class.
 
 ```python
 import versionhq as vhq
@@ -206,34 +219,20 @@ task = vhq.Task(
    callback_kwargs=dict(message="Hi! Here is the result: ")
 )
 
-res = task.execute(context="context to consider")
-
+res = task.execute(context="testing a task function")
 assert isinstance(res, vhq.TaskOutput)
 ```
 
-This will return a `TaskOutput` object that stores response in plane text, JSON, and Pydantic model: `CustomOutput` formats with a callback result, tool output (if given), and evaluation results (if given).
 
-```python
-res == vhq.TaskOutput(
-   task_id=UUID('<TASK UUID>'),
-   raw='{\"test1\":\"random str\", \"test2\":[\"str item 1\", \"str item 2\", \"str item 3\"]}',
-   json_dict={'test1': 'random str', 'test2': ['str item 1', 'str item 2', 'str item 3']},
-   pydantic=<class '__main__.CustomOutput'>,
-   tool_output=None,
-   callback_output='Hi! Here is the result: random str, str item 1, str item 2, str item 3', # returned a plain text summary
-   evaluation=None
-)
-```
-
-### Supervising
+### Supervising agents
 
 To create an agent network with one or more manager agents, designate members using the `is_manager` tag.
 
 ```python
 import versionhq as vhq
 
-agent_a = vhq.Agent(role="agent a", llm="llm-of-your-choice")
-agent_b = vhq.Agent(role="agent b", llm="llm-of-your-choice")
+agent_a = vhq.Agent(role="Member", llm="gpt-4o")
+agent_b = vhq.Agent(role="Leader", llm="gemini-2.0")
 
 task_1 = vhq.Task(
    description="Analyze the client's business model.",
@@ -255,11 +254,9 @@ network =vhq.AgentNetwork(
 )
 res, tg = network.launch()
 
-assert isinstance(res, vhq.TaskOutput)
-assert agent_b.key in task_1.processed_agents
-assert agent_b.key in task_2.processed_agents
-
-assert isinstance(tg, vhq.TaskGraph)
+assert isinstance(res, vhq.NetworkOutput)
+assert not [item for item in task_1.processed_agents if "vhq-Delegated-Agent" == item]
+assert [item for item in task_1.processed_agents if "agent b" == item]
 ```
 
 This will return a list with dictionaries with keys defined in the `ResponseField` of each task.
@@ -423,7 +420,7 @@ Create `.env` file in the project root and add secret vars following `.env.sampl
 
    * Test functions within the files must begin with `test_`.
 
-   * Pytest priorities are `1. playground demo > 2. docs use cases > 3. other features`
+   * Pytest priorities are `1. playground > 2. docs use cases > 3. other features`
 
 
 4. Update `docs` accordingly.
@@ -511,4 +508,4 @@ Common issues and solutions:
 ## Frequently Asked Questions (FAQ)
 **Q. Where can I see if the agent is working?**
 
-A. Visit [playground](https://versi0n.io/playground).
+A. Visit [playground](https://versi0n.io).

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/docs/core/agent/config.md

@@ -11,11 +11,7 @@ By default, when the model provider name is provided, we will select the most co
 ```python
 import versionhq as vhq
 
-agent = vhq.Agent(
-	role="Marketing Analyst",
-	goal="Coping with price competition in saturated markets",
-	llm="gemini-2.0"
-)
+agent = vhq.Agent(role="Marketing Analyst", llm="gemini-2.0")
 ```
 
 <hr/>
@@ -34,7 +30,6 @@ import versionhq as vhq
 
 agent = vhq.Agent(
     role="Marketing Analyst",
-    goal="Coping with increased price competition in saturated markets.",
     respect_context_window=False,
     max_execution_time=60,
     max_rpm=5,
@@ -48,10 +43,10 @@ agent = vhq.Agent(
     )
 
 assert isinstance(agent.llm, vhq.LLM)
-assert agent.llm.temperature == 1
-assert agent.llm.top_p == 0.1
-assert agent.llm.n == 1
-assert agent.llm.stop == "answer"
+assert agent.llm.llm_config["temperature"] == 1
+assert agent.llm.llm_config["top_p"] == 0.1
+assert agent.llm.llm_config["n"] == 1
+assert agent.llm.llm_config["stop"] == "answer"
 ```
 
 <hr>

versionhq-1.2.2.7/docs/core/llm/index.md

@@ -0,0 +1,66 @@
+---
+tags:
+  - Agent Network
+---
+
+# LLM
+
+<class>`class` versionhq.llm.model.<bold>LLM<bold></class>
+
+A Pydantic class to store LLM objects and its task handling rules.
+
+You can specify a model and integration platform from the list. Else, we'll use `gemini` or `gpt` via `LiteLLM` by default.
+
+
+**List of available models**
+
+```python
+"openai": [
+    "gpt-4",
+    "gpt-4o",
+    "gpt-4o-mini",
+    "o1-mini",
+    "o1-preview",
+],
+"gemini": [
+    "gemini/gemini-1.5-flash",
+    "gemini/gemini-1.5-pro",
+    "gemini/gemini-2.0-flash-exp",
+],
+"anthropic": [
+    "claude-3-7-sonnet-latest",
+    "claude-3-5-haiku-latest",
+    "claude-3-5-sonnet-latest",
+    "claude-3-opus-latest",
+],
+"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",
+    "openrouter/databricks/dbrx-instruct",
+],
+"bedrock": [
+    "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/cohere.command-light-text-v14",
+],
+"huggingface": [
+    "huggingface/qwen/qwen2.5-VL-72B-Instruct",
+]
+```

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/docs/core/task/task-strc-response.md

@@ -165,7 +165,7 @@ import versionhq as vhq
 from pydantic import BaseModel
 from typing import Any
 
-# 1. Define and execute a sub task with Pydantic output.
+# 1. Defines a sub task
 class Sub(BaseModel):
     sub1: list[dict[str, Any]]
     sub2: dict[str, Any]
@@ -176,27 +176,28 @@ sub_task = vhq.Task(
 )
 sub_res = sub_task.execute()
 
-# 2. Define a main task, callback function to format the final response.
+# 2. Defines a main task with callbacks
 class Main(BaseModel):
-    main1: list[Any] # <= assume expecting to store Sub object in this field.
-    # error_main1: list[InstanceOf[Sub]]  # as this will trigger 400 error!
+    main1: list[Any] # <= assume expecting to store Sub object.
     main2: dict[str, Any]
 
-def format_response(sub: InstanceOf[Sub], main1: list[Any], main2: dict[str, Any]) -> Main:
-    main1.append(sub)
+def format_response(sub, main1, main2) -> Main:
+    if main1:
+        main1.append(sub)
     main = Main(main1=main1, main2=main2)
     return main
 
-# 3. Execute
+# 3. Executes
 main_task = vhq.Task(
-    description="generate random values that strictly follows the given format",
+    description="generate random values that strictly follows the given format.",
     pydantic_output=Main,
     callback=format_response,
-    callback_kwargs=dict(sub=Sub(sub1=sub_res.pydantic.sub1, sub2=sub_res.pydantic.sub2)),
+    callback_kwargs=dict(sub=sub_res.json_dict),
 )
-res = main_task.execute(context=sub_res.raw) # [Optional] Adding sub_task as a context.
+res = main_task.execute(context=sub_res.raw) # [Optional] Adding sub_task's response as context.
 
-assert [item for item in res.callback_output.main1 if isinstance(item, Sub)]
+assert res.callback_output.main1 is not None
+assert res.callback_output.main2 is not None
 ```
 
 To automate these manual setups, refer to <a href="/core/agent-network">AgentNetwork</a> class.

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/docs/index.md

@@ -25,7 +25,7 @@ A Python framework for agentic orchestration that handles complex task automatio
 
 `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

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/pyproject.toml

@@ -15,7 +15,7 @@ exclude = ["test*", "__pycache__", "*.egg-info"]
 
 [project]
 name = "versionhq"
-version = "1.2.2.5"
+version = "1.2.2.7"
 authors = [{ name = "Kuriko Iwai", email = "kuriko@versi0n.io" }]
 description = "An agentic orchestration framework for building agent networks that handle task automation."
 readme = "README.md"
@@ -46,6 +46,7 @@ dependencies = [
   "composio-core==0.7.0",
   "networkx>=3.4.2",
   "matplotlib>=3.10.0",
+  "boto3>=1.37.1",
 ]
 classifiers = [
   "Programming Language :: Python",

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/src/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.5"
+__version__ = "1.2.2.7"
 __all__ = [
     "Agent",
 

{versionhq-1.2.2.5 → versionhq-1.2.2.7}/src/versionhq/agent/model.py

@@ -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.")
@@ -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.role or not self.goal:
+        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,
         )