fix(agents-api): Fix the typespec bug and regenerate

Signed-off-by: Diwank Tomer <[email protected]>
julep-ai · Aug 20, 2024 · f8111ea · f8111ea
1 parent 9c01928
commit f8111ea
Show file tree

Hide file tree

Showing 42 changed files with 3,540 additions and 339 deletions.
diff --git a/agents-api/agents_api/autogen/Agents.py b/agents-api/agents_api/autogen/Agents.py
@@ -13,6 +13,7 @@
 
 class Agent(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: Annotated[UUID, Field(json_schema_extra={"readOnly": True})]
@@ -59,6 +60,7 @@ class CreateAgentRequest(BaseModel):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     metadata: dict[str, Any] | None = None
@@ -92,6 +94,7 @@ class CreateAgentRequest(BaseModel):
 
 class CreateOrUpdateAgentRequest(CreateAgentRequest):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: UUID
@@ -130,6 +133,7 @@ class PatchAgentRequest(BaseModel):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     metadata: dict[str, Any] | None = None
@@ -167,6 +171,7 @@ class UpdateAgentRequest(BaseModel):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     metadata: dict[str, Any] | None = None

diff --git a/agents-api/agents_api/autogen/Chat.py b/agents-api/agents_api/autogen/Chat.py
@@ -6,7 +6,7 @@
 from typing import Annotated, Literal
 from uuid import UUID
 
-from pydantic import AwareDatetime, BaseModel, ConfigDict, Field
+from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, StrictBool
 
 from .Common import LogitBias
 from .Docs import DocReference
@@ -16,6 +16,7 @@
 
 class BaseChatOutput(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     index: int
@@ -31,6 +32,7 @@ class BaseChatOutput(BaseModel):
 
 class BaseChatResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     usage: CompetionUsage | None = None
@@ -54,6 +56,7 @@ class BaseChatResponse(BaseModel):
 
 class BaseTokenLogProb(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     token: str
@@ -66,6 +69,7 @@ class BaseTokenLogProb(BaseModel):
 
 class ChatInputData(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     messages: Annotated[list[InputChatMLMessage], Field(min_length=1)]
@@ -88,6 +92,7 @@ class ChatOutputChunk(BaseChatOutput):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     delta: InputChatMLMessage
@@ -98,6 +103,7 @@ class ChatOutputChunk(BaseChatOutput):
 
 class ChunkChatResponse(BaseChatResponse):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     choices: list[ChatOutputChunk]
@@ -112,6 +118,7 @@ class CompetionUsage(BaseModel):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     completion_tokens: Annotated[
@@ -136,6 +143,7 @@ class CompetionUsage(BaseModel):
 
 class CompletionResponseFormat(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     type: Literal["text", "json_object"] = "text"
@@ -146,6 +154,7 @@ class CompletionResponseFormat(BaseModel):
 
 class LogProbResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     content: Annotated[list[TokenLogProb] | None, Field(...)]
@@ -156,6 +165,7 @@ class LogProbResponse(BaseModel):
 
 class MessageChatResponse(BaseChatResponse):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     choices: list[SingleChatOutput | MultipleChatOutput]
@@ -170,13 +180,15 @@ class MultipleChatOutput(BaseChatOutput):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     messages: list[InputChatMLMessage]
 
 
 class OpenAISettings(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     frequency_penalty: Annotated[float | None, Field(None, ge=-2.0, le=2.0)]
@@ -203,31 +215,34 @@ class SingleChatOutput(BaseChatOutput):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     message: InputChatMLMessage
 
 
 class TokenLogProb(BaseTokenLogProb):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     top_logprobs: list[BaseTokenLogProb]
 
 
 class ChatInput(ChatInputData):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
-    remember: Annotated[bool, Field(False, json_schema_extra={"readOnly": True})]
+    remember: Annotated[StrictBool, Field(False, json_schema_extra={"readOnly": True})]
     """
     DISABLED: Whether this interaction should form new memories or not (will be enabled in a future release)
     """
-    recall: bool = True
+    recall: StrictBool = True
     """
     Whether previous memories and docs should be recalled or not
     """
-    save: bool = True
+    save: StrictBool = True
     """
     Whether this interaction should be stored in the session history or not
     """
@@ -241,7 +256,7 @@ class ChatInput(ChatInputData):
     """
     Identifier of the model to be used
     """
-    stream: bool = False
+    stream: StrictBool = False
     """
     Indicates if the server should stream the response as it's generated
     """
@@ -305,6 +320,7 @@ class DefaultChatSettings(OpenAISettings):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     repetition_penalty: Annotated[float | None, Field(None, ge=0.0, le=2.0)]
@@ -323,6 +339,7 @@ class DefaultChatSettings(OpenAISettings):
 
 class ChatSettings(DefaultChatSettings):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     model: Annotated[
@@ -335,7 +352,7 @@ class ChatSettings(DefaultChatSettings):
     """
     Identifier of the model to be used
     """
-    stream: bool = False
+    stream: StrictBool = False
     """
     Indicates if the server should stream the response as it's generated
     """

diff --git a/agents-api/agents_api/autogen/Common.py b/agents-api/agents_api/autogen/Common.py
@@ -38,6 +38,7 @@ class Offset(RootModel[int]):
 
 class ResourceCreatedResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: UUID
@@ -56,6 +57,7 @@ class ResourceCreatedResponse(BaseModel):
 
 class ResourceDeletedResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: UUID
@@ -74,6 +76,7 @@ class ResourceDeletedResponse(BaseModel):
 
 class ResourceUpdatedResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: UUID

diff --git a/agents-api/agents_api/autogen/Docs.py b/agents-api/agents_api/autogen/Docs.py
@@ -11,6 +11,7 @@
 
 class BaseDocSearchRequest(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     limit: Annotated[int, Field(10, ge=1, le=100)]
@@ -26,6 +27,7 @@ class CreateDocRequest(BaseModel):
     """
 
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     metadata: dict[str, Any] | None = None
@@ -46,6 +48,7 @@ class CreateDocRequest(BaseModel):
 
 class Doc(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: Annotated[UUID, Field(json_schema_extra={"readOnly": True})]
@@ -71,6 +74,7 @@ class Doc(BaseModel):
 
 class DocOwner(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     id: UUID
@@ -79,6 +83,7 @@ class DocOwner(BaseModel):
 
 class DocReference(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     owner: DocOwner
@@ -96,6 +101,7 @@ class DocReference(BaseModel):
 
 class DocSearchResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     docs: list[DocReference]
@@ -110,6 +116,7 @@ class DocSearchResponse(BaseModel):
 
 class EmbedQueryRequest(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     text: str | list[str]
@@ -120,6 +127,7 @@ class EmbedQueryRequest(BaseModel):
 
 class EmbedQueryResponse(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     vectors: list[list[float]]
@@ -130,6 +138,7 @@ class EmbedQueryResponse(BaseModel):
 
 class HybridDocSearchRequest(BaseDocSearchRequest):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     confidence: Annotated[float, Field(0.5, ge=0.0, le=1.0)]
@@ -152,6 +161,7 @@ class HybridDocSearchRequest(BaseDocSearchRequest):
 
 class Snippet(BaseModel):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     index: int
@@ -160,6 +170,7 @@ class Snippet(BaseModel):
 
 class TextOnlyDocSearchRequest(BaseDocSearchRequest):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     text: str
@@ -170,6 +181,7 @@ class TextOnlyDocSearchRequest(BaseDocSearchRequest):
 
 class VectorDocSearchRequest(BaseDocSearchRequest):
     model_config = ConfigDict(
+        extra="allow",
         populate_by_name=True,
     )
     confidence: Annotated[float, Field(0.5, ge=0.0, le=1.0)]