mirror of
https://github.com/EvolutionAPI/adk-python.git
synced 2025-07-14 01:41:25 -06:00
chore: Updates model_config docstring to explicitly mentioning pydantic.
This is to disambiguate between pydantic model and LLM. PiperOrigin-RevId: 754981617
This commit is contained in:
parent
a930633fad
commit
dbea793cf5
@ -30,6 +30,7 @@ class ActiveStreamingTool(BaseModel):
|
||||
arbitrary_types_allowed=True,
|
||||
extra='forbid',
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
task: Optional[asyncio.Task] = None
|
||||
"""The active task of this streaming tool."""
|
||||
|
@ -50,6 +50,7 @@ class BaseAgent(BaseModel):
|
||||
arbitrary_types_allowed=True,
|
||||
extra='forbid',
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
name: str
|
||||
"""The agent's name.
|
||||
|
@ -110,6 +110,7 @@ class InvocationContext(BaseModel):
|
||||
arbitrary_types_allowed=True,
|
||||
extra="forbid",
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
artifact_service: Optional[BaseArtifactService] = None
|
||||
session_service: BaseSessionService
|
||||
|
@ -53,6 +53,7 @@ class LangGraphAgent(BaseAgent):
|
||||
model_config = ConfigDict(
|
||||
arbitrary_types_allowed=True,
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
graph: CompiledGraph
|
||||
|
||||
|
@ -24,6 +24,7 @@ class LiveRequest(BaseModel):
|
||||
"""Request send to live agents."""
|
||||
|
||||
model_config = ConfigDict(ser_json_bytes='base64', val_json_bytes='base64')
|
||||
"""The pydantic model config."""
|
||||
|
||||
content: Optional[types.Content] = None
|
||||
"""If set, send the content to the model in turn-by-turn mode."""
|
||||
|
@ -37,6 +37,7 @@ class RunConfig(BaseModel):
|
||||
model_config = ConfigDict(
|
||||
extra='forbid',
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
speech_config: Optional[types.SpeechConfig] = None
|
||||
"""Speech configuration for the live agent."""
|
||||
|
@ -26,6 +26,7 @@ class TranscriptionEntry(BaseModel):
|
||||
arbitrary_types_allowed=True,
|
||||
extra='forbid',
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
role: str
|
||||
"""The role that created this data, typically "user" or "model"""
|
||||
|
@ -13,17 +13,16 @@
|
||||
# limitations under the License.
|
||||
|
||||
from enum import Enum
|
||||
from typing import Any
|
||||
from typing import Dict
|
||||
from typing import List
|
||||
from typing import Optional
|
||||
from typing import Any, Dict, List, Optional
|
||||
|
||||
from pydantic import BaseModel
|
||||
from pydantic import ConfigDict
|
||||
from pydantic import Field
|
||||
|
||||
|
||||
class BaseModelWithConfig(BaseModel):
|
||||
model_config = {"extra": "allow"}
|
||||
model_config = ConfigDict(extra="allow")
|
||||
"""The pydantic model config."""
|
||||
|
||||
|
||||
class HttpCredentials(BaseModelWithConfig):
|
||||
|
@ -48,6 +48,7 @@ class Event(LlmResponse):
|
||||
model_config = ConfigDict(
|
||||
extra='forbid', ser_json_bytes='base64', val_json_bytes='base64'
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
# TODO: revert to be required after spark migration
|
||||
invocation_id: str = ''
|
||||
|
@ -27,6 +27,7 @@ class EventActions(BaseModel):
|
||||
"""Represents the actions attached to an event."""
|
||||
|
||||
model_config = ConfigDict(extra='forbid')
|
||||
"""The pydantic model config."""
|
||||
|
||||
skip_summarization: Optional[bool] = None
|
||||
"""If true, it won't call model to summarize function response.
|
||||
|
@ -14,11 +14,9 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from abc import abstractmethod
|
||||
from typing import AsyncGenerator
|
||||
from typing import TYPE_CHECKING
|
||||
from typing import AsyncGenerator, TYPE_CHECKING
|
||||
|
||||
from google.genai import types
|
||||
|
||||
from pydantic import BaseModel
|
||||
from pydantic import ConfigDict
|
||||
|
||||
@ -34,14 +32,13 @@ class BaseLlm(BaseModel):
|
||||
|
||||
Attributes:
|
||||
model: The name of the LLM, e.g. gemini-1.5-flash or gemini-1.5-flash-001.
|
||||
model_config: The model config
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(
|
||||
# This allows us to use arbitrary types in the model. E.g. PIL.Image.
|
||||
arbitrary_types_allowed=True,
|
||||
)
|
||||
"""The model config."""
|
||||
"""The pydantic model config."""
|
||||
|
||||
model: str
|
||||
"""The name of the LLM, e.g. gemini-1.5-flash or gemini-1.5-flash-001."""
|
||||
|
@ -573,7 +573,6 @@ class LiteLlm(BaseLlm):
|
||||
Attributes:
|
||||
model: The name of the LiteLlm model.
|
||||
llm_client: The LLM client to use for the model.
|
||||
model_config: The model config.
|
||||
"""
|
||||
|
||||
llm_client: LiteLLMClient = Field(default_factory=LiteLLMClient)
|
||||
|
@ -37,7 +37,7 @@ class LlmRequest(BaseModel):
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(arbitrary_types_allowed=True)
|
||||
"""The model config."""
|
||||
"""The pydantic model config."""
|
||||
|
||||
model: Optional[str] = None
|
||||
"""The model name."""
|
||||
|
@ -41,7 +41,7 @@ class LlmResponse(BaseModel):
|
||||
"""
|
||||
|
||||
model_config = ConfigDict(extra='forbid')
|
||||
"""The model config."""
|
||||
"""The pydantic model config."""
|
||||
|
||||
content: Optional[types.Content] = None
|
||||
"""The content of the response."""
|
||||
|
@ -38,6 +38,7 @@ class Session(BaseModel):
|
||||
extra='forbid',
|
||||
arbitrary_types_allowed=True,
|
||||
)
|
||||
"""The pydantic model config."""
|
||||
|
||||
id: str
|
||||
"""The unique identifier of the session."""
|
||||
|
Loading…
Reference in New Issue
Block a user