.

Author: smoreinis

examples/tutorials/00_sync/000_hello_acp/project/acp.py

@@ -2,10 +2,12 @@
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 
+
 from agentex.lib.types.task_message_updates import TaskMessageUpdate
 from agentex.types.task_message import TaskMessageContent
 from agentex.types.task_message_content import TextContent
 from agentex.lib.utils.logging import make_logger
+from agentex.lib.utils.model_utils import BaseModel
 
 logger = make_logger(__name__)
 
@@ -16,11 +18,25 @@
 )
 
 
+class StateModel(BaseModel):
+    system_prompt: str
+    model: str
+
 @acp.on_message_send
 async def handle_message_send(
     params: SendMessageParams
 ) -> Union[TaskMessageContent, AsyncGenerator[TaskMessageUpdate, None]]:
+    from agentex.lib import adk
     """Default message handler with streaming support"""
+    # Try to retrieve the state. If it doesn't exist, create it.
+    task_state = await adk.state.get_by_task_and_agent(task_id=params.task.id, agent_id=params.agent.id)
+
+    if not task_state:
+        # If the state doesn't exist, create it.
+        state = StateModel(system_prompt="You are a helpful assistant that can answer questions.", model="gpt-4o-mini")
+        task_state = await adk.state.create(task_id=params.task.id, agent_id=params.agent.id, state=state)
+    else:
+        state = StateModel.model_validate(task_state.state)
     return TextContent(
         author="agent",
         content=f"Hello! I've received your message. Here's a generic response, but in future tutorials we'll see how you can get me to intelligently respond to your message. This is what I heard you say: {params.content.content}",

examples/tutorials/00_sync/010_multiturn/project/acp.py

@@ -1,7 +1,7 @@
 import os
 from typing import AsyncGenerator, Union
 
-from agentex.lib import adk
+
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 from agentex.lib.types.llm_messages import AssistantMessage, LLMConfig, SystemMessage, UserMessage
@@ -26,6 +26,7 @@ class StateModel(BaseModel):
 async def handle_message_send(
     params: SendMessageParams
 ) -> Union[TaskMessageContent, AsyncGenerator[TaskMessageUpdate, None]]:
+    from agentex.lib import adk
     """
     In this tutorial, we'll see how to handle a basic multi-turn conversation without streaming.
     """

src/agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -100,6 +100,21 @@ async def _handle_jsonrpc(self, request: Request):
             data = await request.json()
             rpc_request = JSONRPCRequest(**data)
 
+            # Check if the request is authenticated
+            if os.environ.get("AGENT_API_KEY") is not None:
+                authorization_header = request.headers.get("Authorization")
+                if authorization_header is None:
+                    return JSONRPCResponse(
+                        id=rpc_request.id,
+                        error=JSONRPCError(code=-32601, message="Unauthorized"),
+                    )
+                if authorization_header != f"Bearer {os.environ.get("AGENT_API_KEY")}":
+                    return JSONRPCResponse(
+                        id=rpc_request.id,
+                        error=JSONRPCError(code=-32601, message="Unauthorized"),
+                    )
+
+
             # Check if method is valid first
             try:
                 method = RPCMethod(rpc_request.method)