grasp_agents 0.1.16__py3-none-any.whl → 0.1.17__py3-none-any.whl

grasp_agents/cloud_llm.py

@@ -16,13 +16,13 @@ from tenacity import (
 )
 from typing_extensions import TypedDict
 
-from .data_retrieval.rate_limiter_chunked import (  # type: ignore
-    RateLimiterC,
-    limit_rate_chunked,
-)
 from .http_client import AsyncHTTPClientParams, create_async_http_client
 from .llm import LLM, ConvertT, LLMSettings, SettingsT
 from .memory import MessageHistory
+from .rate_limiting.rate_limiter_chunked import (  # type: ignore
+    RateLimiterC,
+    limit_rate_chunked,
+)
 from .typing.completion import Completion, CompletionChunk
 from .typing.message import AssistantMessage, Conversation
 from .typing.tool import BaseTool, ToolChoice

grasp_agents/openai/openai_llm.py

@@ -3,14 +3,14 @@ from collections.abc import Iterable
 from copy import deepcopy
 from typing import Any, Literal
 
-from openai import AsyncOpenAI
-from openai._types import NOT_GIVEN  # noqa: PLC2701 # type: ignore[import]
 from pydantic import BaseModel
 
-from ..data_retrieval.rate_limiter_chunked import RateLimiterC
+from openai import AsyncOpenAI
+from openai._types import NOT_GIVEN  # type: ignore[import]
 
 from ..cloud_llm import APIProvider, CloudLLM, CloudLLMSettings
 from ..http_client import AsyncHTTPClientParams
+from ..rate_limiting.rate_limiter_chunked import RateLimiterC
 from ..typing.message import AssistantMessage, Conversation
 from ..typing.tool import BaseTool
 from . import (

grasp_agents/{data_retrieval → rate_limiting}/rate_limiter_chunked.py

@@ -3,16 +3,11 @@ import functools
 import logging
 from collections.abc import Callable, Coroutine
 from time import monotonic
-from typing import (
-    Any,
-    Generic,
-    overload,
-)
+from typing import Any, Generic, overload
 
 from tqdm.autonotebook import tqdm
 
 from ..utils import asyncio_gather_with_pbar
-
 from .types import (
     QueryP,
     QueryR,
@@ -54,9 +49,7 @@ class RateLimiterC(Generic[QueryT, QueryR]):
                 if now < self._state.next_request_time:
                     await asyncio.sleep(self._state.next_request_time - now)
                 self._state.next_request_time = monotonic() + 1.01 * 60.0 / self._rpm
-            result = await func_partial(inp)
-
-        return result
+            return await func_partial(inp)
 
     async def process_inputs(
         self,

grasp_agents/{data_retrieval → rate_limiting}/utils.py

@@ -14,11 +14,16 @@ from .types import (
 
 
 def is_bound_method(func: Callable[..., Any], self_candidate: Any) -> bool:
-    return (inspect.ismethod(func) and (func.__self__ is self_candidate)) or hasattr(self_candidate, func.__name__)
+    return (inspect.ismethod(func) and (func.__self__ is self_candidate)) or hasattr(
+        self_candidate, func.__name__
+    )
 
 
 def split_pos_args(
-    call: (RetrievalCallableSingle[QueryT, QueryP, QueryR] | RetrievalCallableList[QueryT, QueryP, QueryR]),
+    call: (
+        RetrievalCallableSingle[QueryT, QueryP, QueryR]
+        | RetrievalCallableList[QueryT, QueryP, QueryR]
+    ),
     args: Sequence[Any],
 ) -> tuple[Any | None, QueryT | list[QueryT], Sequence[Any]]:
     if not args:
@@ -28,12 +33,15 @@ def split_pos_args(
         # Case: Bound instance method with signature (self, inp, *rest)
         if len(args) < 2:
             raise ValueError(
-                "Must pass at least `self` and an input (or a list of inputs) " + "for a bound instance method."
+                "Must pass at least `self` and an input (or a list of inputs) "
+                "for a bound instance method."
             )
         return maybe_self, args[1], args[2:]
     # Case: Standalone function with signature (inp, *rest)
     if not args:
-        raise ValueError("Must pass an input (or a list of inputs) " + "for a standalone function.")
+        raise ValueError(
+            "Must pass an input (or a list of inputs) " + "for a standalone function."
+        )
     return None, args[0], args[1:]
 
 
@@ -53,5 +61,7 @@ def partial_retrieval_callable(
     return wrapper
 
 
-def expected_exec_time_from_max_concurrency_and_rpm(rpm: float, max_concurrency: int) -> float:
+def expected_exec_time_from_max_concurrency_and_rpm(
+    rpm: float, max_concurrency: int
+) -> float:
     return 60.0 / (rpm / max_concurrency)

grasp_agents/utils.py

@@ -1,20 +1,19 @@
 import ast
-import functools
 import asyncio
-from datetime import datetime
+import functools
 import json
 import re
-from logging import getLogger
-from collections.abc import Callable
+from collections.abc import Callable, Coroutine
 from copy import deepcopy
+from datetime import datetime
+from logging import getLogger
 from pathlib import Path
-from typing import Any, TypeVar, cast, Coroutine
-from tqdm.autonotebook import tqdm
+from typing import Any, TypeVar, cast
 
 from pydantic import BaseModel, GetCoreSchemaHandler, TypeAdapter, create_model
 from pydantic.fields import FieldInfo
 from pydantic_core import core_schema
-
+from tqdm.autonotebook import tqdm
 
 logger = getLogger(__name__)
 

grasp_agents-0.1.17.dist-info/METADATA

@@ -0,0 +1,212 @@
+Metadata-Version: 2.4
+Name: grasp_agents
+Version: 0.1.17
+Summary: Grasp Agents Library
+License-File: LICENSE.md
+Requires-Python: <4,>=3.11.4
+Requires-Dist: dotenv>=0.9.9
+Requires-Dist: httpx<1,>=0.27.0
+Requires-Dist: openai<2,>=1.68.2
+Requires-Dist: pydantic>=2
+Requires-Dist: pyyaml>=6.0.2
+Requires-Dist: tenacity>=9.1.2
+Requires-Dist: termcolor<3,>=2.4.0
+Requires-Dist: tqdm<5,>=4.66.2
+Description-Content-Type: text/markdown
+
+# Grasp Agents
+
+<br/>
+<img src="./.assets/grasp.svg" alt="Grasp Agents" width="320" />
+<br/>
+<br/>
+
+[![PyPI version](https://badge.fury.io/py/grasp_agents.svg)](https://badge.fury.io/py/grasp-agents)
+[![License: MIT](https://img.shields.io/badge/license-MIT-yellow?style=flat-square)](https://mit-license.org/)
+[![PyPI downloads](https://img.shields.io/pypi/dm/grasp-agents?style=flat-square)](https://pypi.org/project/grasp-agents/)
+[![GitHub Stars](https://img.shields.io/github/stars/grasp-technologies/grasp-agents?style=social)](https://github.com/grasp-technologies/grasp-agents/stargazers)
+[![GitHub Forks](https://img.shields.io/github/forks/grasp-technologies/grasp-agents?style=social)](https://github.com/grasp-technologies/grasp-agents/network/members)
+
+## Overview
+
+**Grasp Agents** is a modular Python framework for building agentic AI pipelines and applications. It is meant to be minimalistic but functional, allowing for rapid experimentation while keeping full and granular low-level control over prompting, LLM handling, and inter-agent communication by avoiding excessive higher-level abstractions.
+
+## Features
+
+- Clean formulation of agents as generic entities over:
+    * I/O schemas
+    * Agent state
+    * Shared context
+- Transparent implementation of common agentic patterns:
+    * Single-agent loops with an optional "ReAct mode" to enforce reasoning between the tool calls
+    * Workflows (static communication topology), including loops
+    * Agents-as-tools for task delegation
+    * Freeform A2A communication via in-process Actor model
+- Batch processing support outside of agentic loops
+- Simple logging and usage/cost tracking
+
+## Project Structure
+
+- `base_agent.py`, `llm_agent.py`, `comm_agent.py`: Core agent class implementations.
+- `agent_message.py`, `agent_message_pool.py`: Messaging and message pool management.
+- `llm_agent_state.py`: State management for LLM agents.
+- `tool_orchestrator.py`: Orchestration of tools used by agents.
+- `prompt_builder.py`: Tools for constructing prompts.
+- `workflow/`: Modules for defining and managing agent workflows.
+- `cloud_llm.py`, `llm.py`: LLM integration and base LLM functionalities.
+- `openai/`: Modules specific to OpenAI API integration.    
+- `memory.py`: Memory management for agents (currently only message history).
+- `run_context.py`: Context management for agent runs.
+- `usage_tracker.py`: Tracking of API usage and costs.
+- `costs_dict.yaml`: Dictionary for cost tracking (update if needed).
+- `rate_limiting/`: Basic rate limiting tools.
+
+## Quickstart & Installation Variants (UV Package manager)
+
+> **Note:** You can check this sample project code in the [src/grasp_agents/examples/demo/uv](src/grasp_agents/examples/demo/uv) folder. Feel free to copy and paste the code from there to a separate project. There are also [examples](src/grasp_agents/examples/demo/) for other package managers.
+
+#### 1. Prerequisites
+
+Install the [UV Package Manager](https://github.com/astral-sh/uv):
+
+```bash
+curl -LsSf https://astral.sh/uv/install.sh | sh
+```
+
+#### 2. Create Project & Install Dependencies
+
+```bash
+mkdir my-test-uv-app
+cd my-test-uv-app
+uv init .
+```
+
+Create and activate a virtual environment:
+
+```bash
+uv venv
+source .venv/bin/activate
+```
+
+Add and sync dependencies:
+
+```bash
+uv add grasp_agents
+uv sync
+```
+
+#### 3. Example Usage
+
+Ensure you have a `.env` file with your OpenAI and Google AI Studio API keys set
+
+```
+OPENAI_API_KEY=your_openai_api_key
+GOOGLE_AI_STUDIO_API_KEY=your_google_ai_studio_api_key
+```
+
+Create a script, e.g., `problem_recommender.py`:
+
+```python
+import re
+from typing import Any
+from pathlib import Path
+from pydantic import BaseModel, Field
+from dotenv import load_dotenv
+from grasp_agents.typing.tool import BaseTool
+from grasp_agents.typing.io import AgentPayload
+from grasp_agents.run_context import RunContextWrapper
+from grasp_agents.openai.openai_llm import OpenAILLM, OpenAILLMSettings
+from grasp_agents.llm_agent import LLMAgent
+from grasp_agents.grasp_logging import setup_logging
+from grasp_agents.typing.message import Conversation
+
+load_dotenv()
+
+
+# Configure the logger to output to the console and/or a file
+setup_logging(
+    logs_file_path="grasp_agents_demo.log",
+    logs_config_path=Path().cwd() / "configs/logging/default.yaml",
+)
+
+sys_prompt_react = """
+Your task is to suggest an exciting stats problem to a student. 
+Ask the student about their education, interests, and preferences, then suggest a problem tailored to them. 
+
+# Instructions
+* Ask questions one by one.
+* Provide your thinking before asking a question and after receiving a reply.
+* The problem must be enclosed in <PROBLEM> tags.
+"""
+
+
+class TeacherQuestion(BaseModel):
+    question: str = Field(..., description="The question to ask the student.")
+
+StudentReply = str
+
+
+class AskStudentTool(BaseTool[TeacherQuestion, StudentReply, Any]):
+    name: str = "ask_student_tool"
+    description: str = "Ask the student a question and get their reply."
+    in_schema: type[TeacherQuestion] = TeacherQuestion
+    out_schema: type[StudentReply] = StudentReply
+
+    async def run(
+        self, inp: TeacherQuestion, ctx: RunContextWrapper[Any] | None = None
+    ) -> StudentReply:
+        return input(inp.question)
+
+
+class FinalResponse(AgentPayload):
+    problem: str
+
+
+teacher = LLMAgent[Any, FinalResponse, None](
+    agent_id="teacher",
+    llm=OpenAILLM(
+        model_name="gpt-4.1",
+        api_provider="openai",
+        llm_settings=OpenAILLMSettings(temperature=0.1),
+    ),
+    tools=[AskStudentTool()],
+    max_turns=20,
+    react_mode=True,
+    sys_prompt=sys_prompt_react,
+    out_schema=FinalResponse,
+    set_state_strategy="reset",
+)
+
+
+@teacher.tool_call_loop_exit_handler
+def exit_tool_call_loop(conversation: Conversation, ctx, **kwargs) -> None:
+    message_text = conversation[-1].content
+
+    return re.search(r"<PROBLEM>", message_text)
+
+
+@teacher.parse_output_handler
+def parse_output(conversation: Conversation, ctx, **kwargs) -> FinalResponse:
+    message_text = conversation[-1].content
+    matches = re.findall(r"<PROBLEM>(.*?)</PROBLEM>", message_text, re.DOTALL)
+
+    return FinalResponse(problem=matches[0])
+
+
+async def main():
+    ctx = RunContextWrapper(print_messages=True)
+    out = await teacher.run(ctx=ctx)
+    print(out.payloads[0].problem)
+    print(ctx.usage_tracker.total_usage)
+
+
+asyncio.run(main())
+```
+
+Run your script:
+
+```bash
+uv run problem_recommender.py
+```
+
+You can find more examples in [src/grasp_agents/examples/notebooks/agents_demo.ipynb](src/grasp_agents/examples/notebooks/agents_demo.ipynb).

{grasp_agents-0.1.16.dist-info → grasp_agents-0.1.17.dist-info}/RECORD

@@ -1,7 +1,7 @@
 grasp_agents/agent_message.py,sha256=Z-czIHNSe7eAo5r6q2Zu10HFpUjQjVbayyglGyyj4Lw,911
 grasp_agents/agent_message_pool.py,sha256=4O4xz-aZ_I5m3iLAUiMAQcVn5AGRP4daUM8XME03Xsw,3250
 grasp_agents/base_agent.py,sha256=4mNMyWdt9MTzg64JhrVjcU_FSYDj0ED_AKpRZXCRVb4,1870
-grasp_agents/cloud_llm.py,sha256=q72bDyC9bxQdiO552LiOYr58B9T3bMAJu0TGUZH4Lfk,13016
+grasp_agents/cloud_llm.py,sha256=JJmIUu8SOJ4TElQ4B19mPgXJ1SiapZdPFFhA2YVRKRw,13015
 grasp_agents/comm_agent.py,sha256=ukkT0zXgE8-yt63fyi3MQ_3Q0QjwIDHKz4MbHt9FBxw,7323
 grasp_agents/costs_dict.yaml,sha256=EW6XxRXLZobMwQEEiUNYALbDzfbZFb2zEVCaTSAqYjw,2334
 grasp_agents/grasp_logging.py,sha256=H1GYhXdQvVkmauFDZ-KDwvVmPQHZUUm9sRqX_ObK2xI,1111
@@ -15,18 +15,18 @@ grasp_agents/prompt_builder.py,sha256=rYVIY4adJwBitjrTYvpEh5x8C7cLbIiXxT1F-vQuvE
 grasp_agents/run_context.py,sha256=hyETO3-p0azPFws75kX6rrUDLf58Ar6jmyt6TQ5Po78,2589
 grasp_agents/tool_orchestrator.py,sha256=6VX8FGYeUHiSt9GpUjOga7KGP55Kzs1jEiNcOZysIAo,5501
 grasp_agents/usage_tracker.py,sha256=5YuN6hpg6HASdg-hOylgWzhCiORmDMnZuQtbISfhm_4,3378
-grasp_agents/utils.py,sha256=tacUTUnPqz8qUvJxGpGhcdXOyvZJOzfblCrtzqfkCj8,5870
-grasp_agents/data_retrieval/__init__.py,sha256=KRgtF_E7R3YfA2cpYcFcZ7wycV0pWVJ0xRQC7YhiIEQ,158
-grasp_agents/data_retrieval/rate_limiter_chunked.py,sha256=NPqYrWwKTx1lim_zlhWar5wDwFz1cA-b6JOzT10kOtE,5843
-grasp_agents/data_retrieval/types.py,sha256=JbLYJC-gmzcHH_4-YNTz9IcIwVpcpDyDGvljxNznf5k,1389
-grasp_agents/data_retrieval/utils.py,sha256=D3Bkq6-9gF7ubearjZwZaTt_u2-sM3JDlGQD9HmJ3rQ,1917
+grasp_agents/utils.py,sha256=-OdCs27FKek8owGbwmAVWwLXDgEzoTu-B5_EwRLEsz4,5869
 grasp_agents/openai/__init__.py,sha256=qN8HMAatSJKOsA6v-JwakMYguwkswCVHqrmK1gFy9wI,3096
 grasp_agents/openai/completion_converters.py,sha256=lX9h1kaGAo5ttsl-4V7l4x8IpjxJaJJtyU2cKu3-EOc,1871
 grasp_agents/openai/content_converters.py,sha256=6GI0D7xJalzsiawAJOyCUzTJTo0NQdpv87YKmfN0LYQ,2631
 grasp_agents/openai/converters.py,sha256=DBXBxow9oRG6pc8inpZBLiuUqHzVfpscmHFpN9bAdvc,5276
 grasp_agents/openai/message_converters.py,sha256=KjF6FbXzwlWdM-1YT3cswUV-74sjiwOhLFPMY4sJ5Xk,4593
-grasp_agents/openai/openai_llm.py,sha256=rHeix_lSH8mogPfHjgrsaYMrHBMt5q9k-4x33_-zwyA,6149
+grasp_agents/openai/openai_llm.py,sha256=dscGlVhH4v1yAw4NRPgJdww9toOoMRpIqA6HD4IGWOs,6132
 grasp_agents/openai/tool_converters.py,sha256=KhWRETkjhjocISUo_HBZ8QfBiyTOoC5WurPNAR4BYxc,1027
+grasp_agents/rate_limiting/__init__.py,sha256=KRgtF_E7R3YfA2cpYcFcZ7wycV0pWVJ0xRQC7YhiIEQ,158
+grasp_agents/rate_limiting/rate_limiter_chunked.py,sha256=jp6bqWGq2qYAxbmyU_bAGeBXM_nEmSrR6VabDutcTA0,5800
+grasp_agents/rate_limiting/types.py,sha256=JbLYJC-gmzcHH_4-YNTz9IcIwVpcpDyDGvljxNznf5k,1389
+grasp_agents/rate_limiting/utils.py,sha256=yrqboDWWqeekDcs0xaeOPOvelhgAlywDpQpscpPqlhg,1995
 grasp_agents/typing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 grasp_agents/typing/completion.py,sha256=_KDLx3Gtz7o-pEZrvAFgCZwDmkr2oQkxrL-2LSXHHsw,657
 grasp_agents/typing/content.py,sha256=13nLNZqZgtpo9sM0vCRQmZ4bQjjZqUSElMQOwjL7bO8,3651
@@ -38,7 +38,7 @@ grasp_agents/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 grasp_agents/workflow/looped_agent.py,sha256=YBOgOIvy3_NwKvEoGgzQJ2fY9SNG66MQk6obSBGWvCc,3896
 grasp_agents/workflow/sequential_agent.py,sha256=yDt2nA-b1leVByD8jsKrWD6bHe0o9z33jrOJGOLwbyk,2004
 grasp_agents/workflow/workflow_agent.py,sha256=9U94IQ39Vb1W_5u8aoqHb65ikdarEhEJkexDz8xwHD4,2294
-grasp_agents-0.1.16.dist-info/METADATA,sha256=s8QElgMOrmr0ZXlfWgRvPbz-kyfDtUBPnCu2EPCUsBY,4254
-grasp_agents-0.1.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-grasp_agents-0.1.16.dist-info/licenses/LICENSE.md,sha256=Kfeo0gdlLS6tLQiWwO9UWhjp9-f93a5kShSiBp2FG-c,1201
-grasp_agents-0.1.16.dist-info/RECORD,,
+grasp_agents-0.1.17.dist-info/METADATA,sha256=YG_CGwTAOAQFAOA_Aqi6pqwF_ODYtzY_nQEgf2IRMFM,7001
+grasp_agents-0.1.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+grasp_agents-0.1.17.dist-info/licenses/LICENSE.md,sha256=Kfeo0gdlLS6tLQiWwO9UWhjp9-f93a5kShSiBp2FG-c,1201
+grasp_agents-0.1.17.dist-info/RECORD,,

grasp_agents-0.1.16.dist-info/METADATA

@@ -1,151 +0,0 @@
-Metadata-Version: 2.4
-Name: grasp_agents
-Version: 0.1.16
-Summary: Grasp Agents Library
-License-File: LICENSE.md
-Requires-Python: <4,>=3.11.4
-Requires-Dist: dotenv>=0.9.9
-Requires-Dist: httpx<1,>=0.27.0
-Requires-Dist: openai<2,>=1.68.2
-Requires-Dist: pydantic>=2
-Requires-Dist: pyyaml>=6.0.2
-Requires-Dist: tenacity>=9.1.2
-Requires-Dist: termcolor<3,>=2.4.0
-Requires-Dist: tqdm<5,>=4.66.2
-Description-Content-Type: text/markdown
-
-# Grasp Agents
-
-<br/>
-<img src="./.assets/grasp.svg" alt="Grasp Agents" width="320" />
-<br/>
-<br/>
-
-[![PyPI version](https://badge.fury.io/py/grasp_agents.svg)](https://badge.fury.io/py/grasp-agents)
-[![License: MIT](https://img.shields.io/badge/license-MIT-yellow?style=flat-square)](https://mit-license.org/)
-[![PyPI downloads](https://img.shields.io/pypi/dm/grasp-agents?style=flat-square)](https://pypi.org/project/grasp-agents/)
-[![GitHub Stars](https://img.shields.io/github/stars/grasp-technologies/grasp-agents?style=social)](https://github.com/grasp-technologies/grasp-agents/stargazers)
-[![GitHub Forks](https://img.shields.io/github/forks/grasp-technologies/grasp-agents?style=social)](https://github.com/grasp-technologies/grasp-agents/network/members)
-
-## Overview
-
-**Grasp Agents** is a modular Python framework for building agentic AI pipelines and applications. It provides reusable agent classes, message handling, LLM integration, memory, and orchestration utilities. The framework is designed for flexibility, composability, and clarity, enabling rapid prototyping and robust development of multi-agent systems.
-
-## Features
-
-- Modular agent base classes
-- Message and memory management
-- LLM and tool orchestration
-- Logging and usage tracking
-- Extensible architecture
-
-## Project Structure
-
-- `src/grasp_agents/` — Core framework modules
-  - `base_agent.py`, `llm_agent.py`, `comm_agent.py`: Agent classes
-  - `agent_message.py`, `agent_message_pool.py`: Messaging
-  - `memory.py`: Memory management
-  - `cloud_llm.py`, `llm.py`: LLM integration
-  - `tool_orchestrator.py`: Tool orchestration
-  - `usage_tracker.py`, `grasp_logging.py`: Usage and logging
-  - `data_retrieval/`, `openai/`, `typing/`, `workflow/`: Extensions and utilities
-- `configs/` — Configuration files
-- `data/` — Logs and datasets
-
-## Quickstart & Installation Variants (UV Package manager)
-
-> **Note:** You can check this sample project code in the [src/grasp_agents/examples/demo/uv](src/grasp_agents/examples/demo/uv) folder. Feel free to copy and paste the code from there to a separate project. There are also [examples](src/grasp_agents/examples/demo/) for other package managers.
-
-#### 1. Prerequisites
-
-Install the [UV Package Manager](https://github.com/astral-sh/uv):
-
-```bash
-curl -LsSf https://astral.sh/uv/install.sh | sh
-```
-
-#### 2. Create Project & Install Dependencies
-
-```bash
-mkdir my-test-uv-app
-cd my-test-uv-app
-uv init .
-```
-
-Create and activate a virtual environment:
-
-```bash
-uv venv
-source .venv/bin/activate
-```
-
-Add and sync dependencies:
-
-```bash
-uv add grasp_agents
-uv sync
-```
-
-#### 3. Example Usage
-
-Create a file, e.g., `hello.py`:
-
-Ensure you have a `.env` file with your OpenAI and Google AI Studio API keys set
-
-```
-OPENAI_API_KEY=your_openai_api_key
-GOOGLE_AI_STUDIO_API_KEY=your_google_ai_studio_api_key
-```
-
-```python
-import asyncio
-from typing import Any
-
-from grasp_agents.llm_agent import LLMAgent
-from grasp_agents.openai.openai_llm import (
-    OpenAILLM,
-    OpenAILLMSettings,
-)
-from grasp_agents.typing.io import (
-    AgentPayload,
-)
-from grasp_agents.run_context import RunContextWrapper
-
-from dotenv import load_dotenv
-
-load_dotenv()
-
-class Response(AgentPayload):
-    response: str
-
-
-chatbot = LLMAgent[Any, Response, None](
-    agent_id="chatbot",
-    llm=OpenAILLM(
-        model_name="gpt-4o",
-        llm_settings=OpenAILLMSettings(),
-    ),
-    sys_prompt=None,
-    out_schema=Response,
-)
-
-
-@chatbot.parse_output_handler
-def output_handler(conversation, ctx, **kwargs) -> Response:
-    return Response(response=conversation[-1].content)
-
-
-async def main():
-    ctx = RunContextWrapper(print_messages=True)
-    out = await chatbot.run("Hello, agent!", ctx=ctx)
-    print(out.payloads[0].response)
-
-
-asyncio.run(main())
-```
-
-Run your script:
-
-```bash
-uv run hello.py
-```