Codegen for all new typespec types

agents-api/agents_api/activities/embed_docs.py

@@ -1,11 +1,11 @@
 from pydantic import UUID4
 from temporalio import activity
+
+from agents_api.embed_models_registry import EmbeddingModel
 from agents_api.env import embedding_model_id
 from agents_api.models.docs.embed_docs import (
     embed_docs_snippets_query,
 )
-from agents_api.embed_models_registry import EmbeddingModel
-
 
 snippet_embed_instruction = "Encode this passage for retrieval: "
 

agents-api/agents_api/activities/logger.py

@@ -1,6 +1,5 @@
 import logging
 
-
 logger = logging.getLogger(__name__)
 h = logging.StreamHandler()
 fmt = logging.Formatter("[%(asctime)s/%(levelname)s] - %(message)s")

agents-api/agents_api/activities/mem_mgmt.py

@@ -7,7 +7,6 @@
 from ..clients.model import julep_client
 from .types import ChatML, MemoryManagementTaskArgs
 
-
 example_previous_memory = """
 Speaker 1: Composes and listens to music. Likes to buy basketball shoes but doesn't wear them often.
 """.strip()

agents-api/agents_api/activities/summarization.py

@@ -1,22 +1,24 @@
 #!/usr/bin/env python3
 
 import asyncio
-from uuid import UUID
-from typing import Callable
 from textwrap import dedent
-from temporalio import activity
+from typing import Callable
+from uuid import UUID
+
 from litellm import acompletion
+from temporalio import activity
+
+from agents_api.common.protocol.entries import Entry
 from agents_api.models.entry.entries_summarization import (
-    get_toplevel_entries_query,
     entries_summarization_query,
+    get_toplevel_entries_query,
 )
-from agents_api.common.protocol.entries import Entry
-from ..model_registry import LOCAL_MODELS
-from ..env import model_inference_url, model_api_key, summarization_model_name
 from agents_api.rec_sum.entities import get_entities
 from agents_api.rec_sum.summarize import summarize_messages
 from agents_api.rec_sum.trim import trim_messages
 
+from ..env import model_api_key, model_inference_url, summarization_model_name
+from ..model_registry import LOCAL_MODELS
 
 example_previous_memory = """
 Speaker 1: Composes and listens to music. Likes to buy basketball shoes but doesn't wear them often.

agents-api/agents_api/activities/task_steps/__init__.py

@@ -1,34 +1,32 @@
 import asyncio
+from uuid import uuid4
+
+from openai.types.chat.chat_completion import ChatCompletion
 
 # import celpy
 from simpleeval import simple_eval
-from openai.types.chat.chat_completion import ChatCompletion
 from temporalio import activity
-from uuid import uuid4
 
 from ...autogen.openapi_model import (
-    PromptWorkflowStep,
-    # EvaluateWorkflowStep,
-    YieldWorkflowStep,
     # ToolCallWorkflowStep,
     # ErrorWorkflowStep,
     IfElseWorkflowStep,
     InputChatMLMessage,
+    PromptWorkflowStep,
+    # EvaluateWorkflowStep,
+    YieldWorkflowStep,
 )
-
+from ...clients.worker.types import ChatML
 from ...common.protocol.tasks import (
     StepContext,
     TransitionInfo,
 )
-
 from ...common.utils.template import render_template
-
 from ...models.execution.create_execution_transition import (
     create_execution_transition_query,
 )
-from ...routers.sessions.session import llm_generate
 from ...routers.sessions.protocol import Settings
-from ...clients.worker.types import ChatML
+from ...routers.sessions.session import llm_generate
 
 
 @activity.defn

agents-api/agents_api/activities/truncation.py

@@ -1,9 +1,11 @@
 from uuid import UUID
+
 from temporalio import activity
-from agents_api.models.entry.entries_summarization import get_toplevel_entries_query
-from agents_api.models.entry.delete_entries import delete_entries
+
 from agents_api.autogen.openapi_model import Role
 from agents_api.common.protocol.entries import Entry
+from agents_api.models.entry.delete_entries import delete_entries
+from agents_api.models.entry.entries_summarization import get_toplevel_entries_query
 
 
 def get_extra_entries(messages: list[Entry], token_count_threshold: int) -> list[UUID]:

agents-api/agents_api/autogen/Agents.py

@@ -0,0 +1,172 @@
+# generated by datamodel-codegen:
+#   filename:  openapi-0.4.0.yaml
+
+from __future__ import annotations
+
+from typing import Annotated, Any
+from uuid import UUID
+
+from pydantic import AwareDatetime, BaseModel, ConfigDict, Field
+
+from .Chat import GenerationPresetSettings, OpenAISettings, VLLMSettings
+
+
+class Agent(BaseModel):
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    id: Annotated[UUID, Field(json_schema_extra={"readOnly": True})]
+    metadata: dict[str, Any] | None = None
+    created_at: Annotated[AwareDatetime, Field(json_schema_extra={"readOnly": True})]
+    """
+    When this resource was created as UTC date-time
+    """
+    updated_at: Annotated[AwareDatetime, Field(json_schema_extra={"readOnly": True})]
+    """
+    When this resource was updated as UTC date-time
+    """
+    name: Annotated[
+        str,
+        Field(
+            "",
+            pattern="^[\\p{L}\\p{Nl}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]+[\\p{ID_Start}\\p{Mn}\\p{Mc}\\p{Nd}\\p{Pc}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]*$",
+        ),
+    ]
+    """
+    Name of the agent
+    """
+    about: str = ""
+    """
+    About the agent
+    """
+    model: str = ""
+    """
+    Model name to use (gpt-4-turbo, gemini-nano etc)
+    """
+    instructions: str | list[str] = ""
+    """
+    Instructions for the agent
+    """
+    default_settings: (
+        GenerationPresetSettings | OpenAISettings | VLLMSettings | None
+    ) = None
+    """
+    Default settings for all sessions created by this agent
+    """
+
+
+class CreateAgentRequest(BaseModel):
+    """
+    Payload for creating a agent (and associated documents)
+    """
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    metadata: dict[str, Any] | None = None
+    name: Annotated[
+        str,
+        Field(
+            "",
+            pattern="^[\\p{L}\\p{Nl}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]+[\\p{ID_Start}\\p{Mn}\\p{Mc}\\p{Nd}\\p{Pc}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]*$",
+        ),
+    ]
+    """
+    Name of the agent
+    """
+    about: str = ""
+    """
+    About the agent
+    """
+    model: str = ""
+    """
+    Model name to use (gpt-4-turbo, gemini-nano etc)
+    """
+    instructions: str | list[str] = ""
+    """
+    Instructions for the agent
+    """
+    default_settings: (
+        GenerationPresetSettings | OpenAISettings | VLLMSettings | None
+    ) = None
+    """
+    Default settings for all sessions created by this agent
+    """
+
+
+class PatchAgentRequest(BaseModel):
+    """
+    Payload for patching a agent
+    """
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    metadata: dict[str, Any] | None = None
+    name: Annotated[
+        str,
+        Field(
+            "",
+            pattern="^[\\p{L}\\p{Nl}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]+[\\p{ID_Start}\\p{Mn}\\p{Mc}\\p{Nd}\\p{Pc}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]*$",
+        ),
+    ]
+    """
+    Name of the agent
+    """
+    about: str = ""
+    """
+    About the agent
+    """
+    model: str = ""
+    """
+    Model name to use (gpt-4-turbo, gemini-nano etc)
+    """
+    instructions: str | list[str] = ""
+    """
+    Instructions for the agent
+    """
+    default_settings: (
+        GenerationPresetSettings | OpenAISettings | VLLMSettings | None
+    ) = None
+    """
+    Default settings for all sessions created by this agent
+    """
+
+
+class UpdateAgentRequest(BaseModel):
+    """
+    Payload for updating a agent
+    """
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+    )
+    metadata: dict[str, Any] | None = None
+    name: Annotated[
+        str,
+        Field(
+            "",
+            pattern="^[\\p{L}\\p{Nl}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]+[\\p{ID_Start}\\p{Mn}\\p{Mc}\\p{Nd}\\p{Pc}\\p{Pattern_Syntax}\\p{Pattern_White_Space}]*$",
+        ),
+    ]
+    """
+    Name of the agent
+    """
+    about: str = ""
+    """
+    About the agent
+    """
+    model: str = ""
+    """
+    Model name to use (gpt-4-turbo, gemini-nano etc)
+    """
+    instructions: str | list[str] = ""
+    """
+    Instructions for the agent
+    """
+    default_settings: (
+        GenerationPresetSettings | OpenAISettings | VLLMSettings | None
+    ) = None
+    """
+    Default settings for all sessions created by this agent
+    """