fix(manual): update lib/ references to use the alpha namespace

Author: ashwinb

src/llama_stack_client/__init__.py

@@ -41,8 +41,8 @@
 from .lib.agents.agent import Agent
 from .lib.agents.event_logger import EventLogger as AgentEventLogger
 from .lib.inference.event_logger import EventLogger as InferenceEventLogger
+from .types.alpha.agents.turn_create_params import Document
 from .types.shared_params.document import Document as RAGDocument
-from .types.agents.turn_create_params import Document
 
 __all__ = [
     "types",

src/llama_stack_client/lib/agents/agent.py

@@ -7,13 +7,14 @@
 from typing import Any, AsyncIterator, Callable, Iterator, List, Optional, Tuple, Union
 
 from llama_stack_client import LlamaStackClient
-from llama_stack_client.types import ToolResponseMessage, ToolResponseParam, UserMessage
-from llama_stack_client.types.agent_create_params import AgentConfig
-from llama_stack_client.types.agents.agent_turn_response_stream_chunk import (
+from llama_stack_client.types import ToolResponseMessage, UserMessage
+from llama_stack_client.types.alpha import ToolResponseParam
+from llama_stack_client.types.alpha.agent_create_params import AgentConfig
+from llama_stack_client.types.alpha.agents.agent_turn_response_stream_chunk import (
     AgentTurnResponseStreamChunk,
 )
-from llama_stack_client.types.agents.turn import CompletionMessage, Turn
-from llama_stack_client.types.agents.turn_create_params import Document, Toolgroup
+from llama_stack_client.types.alpha.agents.turn import CompletionMessage, Turn
+from llama_stack_client.types.alpha.agents.turn_create_params import Document, Toolgroup
 from llama_stack_client.types.shared.tool_call import ToolCall
 from llama_stack_client.types.shared_params.agent_config import ToolConfig
 from llama_stack_client.types.shared_params.response_format import ResponseFormat
@@ -203,7 +204,7 @@ def __init__(
         self.initialize()
 
     def initialize(self) -> None:
-        agentic_system_create_response = self.client.agents.create(
+        agentic_system_create_response = self.client.alpha.agents.create(
             agent_config=self.agent_config,
             extra_headers=self.extra_headers,
         )
@@ -214,7 +215,7 @@ def initialize(self) -> None:
                 self.builtin_tools[tool.identifier] = tg.get("args", {}) if isinstance(tg, dict) else {}
 
     def create_session(self, session_name: str) -> str:
-        agentic_system_create_session_response = self.client.agents.session.create(
+        agentic_system_create_session_response = self.client.alpha.agents.session.create(
             agent_id=self.agent_id,
             session_name=session_name,
             extra_headers=self.extra_headers,
@@ -322,7 +323,7 @@ def _create_turn_streaming(
         n_iter = 0
 
         # 1. create an agent turn
-        turn_response = self.client.agents.turn.create(
+        turn_response = self.client.alpha.agents.turn.create(
             agent_id=self.agent_id,
             # use specified session_id or last session created
             session_id=session_id or self.session_id[-1],
@@ -361,7 +362,7 @@ def _create_turn_streaming(
                     tool_responses = self._run_tool_calls(tool_calls)
 
                     # pass it to next iteration
-                    turn_response = self.client.agents.turn.resume(
+                    turn_response = self.client.alpha.agents.turn.resume(
                         agent_id=self.agent_id,
                         session_id=session_id or self.session_id[-1],
                         turn_id=turn_id,
@@ -468,7 +469,7 @@ async def initialize(self) -> None:
         if self._agent_id:
             return
 
-        agentic_system_create_response = await self.client.agents.create(
+        agentic_system_create_response = await self.client.alpha.agents.create(
             agent_config=self.agent_config,
         )
         self._agent_id = agentic_system_create_response.agent_id
@@ -478,7 +479,7 @@ async def initialize(self) -> None:
 
     async def create_session(self, session_name: str) -> str:
         await self.initialize()
-        agentic_system_create_session_response = await self.client.agents.session.create(
+        agentic_system_create_session_response = await self.client.alpha.agents.session.create(
             agent_id=self.agent_id,
             session_name=session_name,
             extra_headers=self.extra_headers,
@@ -558,7 +559,7 @@ async def _create_turn_streaming(
         n_iter = 0
 
         # 1. create an agent turn
-        turn_response = await self.client.agents.turn.create(
+        turn_response = await self.client.alpha.agents.turn.create(
             agent_id=self.agent_id,
             # use specified session_id or last session created
             session_id=session_id or self.session_id[-1],
@@ -596,7 +597,7 @@ async def _create_turn_streaming(
                     tool_responses = await self._run_tool_calls(tool_calls)
 
                     # pass it to next iteration
-                    turn_response = await self.client.agents.turn.resume(
+                    turn_response = await self.client.alpha.agents.turn.resume(
                         agent_id=self.agent_id,
                         session_id=session_id or self.session_id[-1],
                         turn_id=turn_id,

src/llama_stack_client/lib/agents/client_tool.py

@@ -19,7 +19,8 @@
     Union,
 )
 
-from llama_stack_client.types import CompletionMessage, Message, ToolResponse
+from llama_stack_client.types import CompletionMessage, Message
+from llama_stack_client.types.alpha import ToolResponse
 from llama_stack_client.types.tool_def_param import Parameter, ToolDefParam
 
 

src/llama_stack_client/lib/agents/tool_parser.py

@@ -7,7 +7,7 @@
 from abc import abstractmethod
 from typing import List
 
-from llama_stack_client.types.agents.turn import CompletionMessage
+from llama_stack_client.types.alpha.agents.turn import CompletionMessage
 from llama_stack_client.types.shared.tool_call import ToolCall