66
77from .._models import BaseModel
88from .agents .api_meta import APIMeta
9+ from .api_agent_model import APIAgentModel
910from .agents .api_links import APILinks
1011from .api_knowledge_base import APIKnowledgeBase
1112from .api_retrieval_method import APIRetrievalMethod
1718 "AgentChatbot" ,
1819 "AgentChatbotIdentifier" ,
1920 "AgentDeployment" ,
20- "AgentModel" ,
21- "AgentModelAgreement" ,
22- "AgentModelVersion" ,
2321 "AgentTemplate" ,
2422 "AgentTemplateGuardrail" ,
25- "AgentTemplateModel" ,
26- "AgentTemplateModelAgreement" ,
27- "AgentTemplateModelVersion" ,
2823]
2924
3025
@@ -74,140 +69,12 @@ class AgentDeployment(BaseModel):
7469 visibility : Optional [APIDeploymentVisibility ] = None
7570
7671
77- class AgentModelAgreement (BaseModel ):
78- description : Optional [str ] = None
79-
80- name : Optional [str ] = None
81-
82- url : Optional [str ] = None
83-
84- uuid : Optional [str ] = None
85-
86-
87- class AgentModelVersion (BaseModel ):
88- major : Optional [int ] = None
89-
90- minor : Optional [int ] = None
91-
92- patch : Optional [int ] = None
93-
94-
95- class AgentModel (BaseModel ):
96- agreement : Optional [AgentModelAgreement ] = None
97-
98- created_at : Optional [datetime ] = None
99-
100- inference_name : Optional [str ] = None
101-
102- inference_version : Optional [str ] = None
103-
104- is_foundational : Optional [bool ] = None
105-
106- metadata : Optional [object ] = None
107-
108- name : Optional [str ] = None
109-
110- parent_uuid : Optional [str ] = None
111-
112- provider : Optional [Literal ["MODEL_PROVIDER_DIGITALOCEAN" , "MODEL_PROVIDER_ANTHROPIC" , "MODEL_PROVIDER_OPENAI" ]] = (
113- None
114- )
115-
116- updated_at : Optional [datetime ] = None
117-
118- upload_complete : Optional [bool ] = None
119-
120- url : Optional [str ] = None
121-
122- usecases : Optional [
123- List [
124- Literal [
125- "MODEL_USECASE_UNKNOWN" ,
126- "MODEL_USECASE_AGENT" ,
127- "MODEL_USECASE_FINETUNED" ,
128- "MODEL_USECASE_KNOWLEDGEBASE" ,
129- "MODEL_USECASE_GUARDRAIL" ,
130- "MODEL_USECASE_REASONING" ,
131- "MODEL_USECASE_SERVERLESS" ,
132- ]
133- ]
134- ] = None
135-
136- uuid : Optional [str ] = None
137-
138- version : Optional [AgentModelVersion ] = None
139-
140-
14172class AgentTemplateGuardrail (BaseModel ):
14273 priority : Optional [int ] = None
14374
14475 uuid : Optional [str ] = None
14576
14677
147- class AgentTemplateModelAgreement (BaseModel ):
148- description : Optional [str ] = None
149-
150- name : Optional [str ] = None
151-
152- url : Optional [str ] = None
153-
154- uuid : Optional [str ] = None
155-
156-
157- class AgentTemplateModelVersion (BaseModel ):
158- major : Optional [int ] = None
159-
160- minor : Optional [int ] = None
161-
162- patch : Optional [int ] = None
163-
164-
165- class AgentTemplateModel (BaseModel ):
166- agreement : Optional [AgentTemplateModelAgreement ] = None
167-
168- created_at : Optional [datetime ] = None
169-
170- inference_name : Optional [str ] = None
171-
172- inference_version : Optional [str ] = None
173-
174- is_foundational : Optional [bool ] = None
175-
176- metadata : Optional [object ] = None
177-
178- name : Optional [str ] = None
179-
180- parent_uuid : Optional [str ] = None
181-
182- provider : Optional [Literal ["MODEL_PROVIDER_DIGITALOCEAN" , "MODEL_PROVIDER_ANTHROPIC" , "MODEL_PROVIDER_OPENAI" ]] = (
183- None
184- )
185-
186- updated_at : Optional [datetime ] = None
187-
188- upload_complete : Optional [bool ] = None
189-
190- url : Optional [str ] = None
191-
192- usecases : Optional [
193- List [
194- Literal [
195- "MODEL_USECASE_UNKNOWN" ,
196- "MODEL_USECASE_AGENT" ,
197- "MODEL_USECASE_FINETUNED" ,
198- "MODEL_USECASE_KNOWLEDGEBASE" ,
199- "MODEL_USECASE_GUARDRAIL" ,
200- "MODEL_USECASE_REASONING" ,
201- "MODEL_USECASE_SERVERLESS" ,
202- ]
203- ]
204- ] = None
205-
206- uuid : Optional [str ] = None
207-
208- version : Optional [AgentTemplateModelVersion ] = None
209-
210-
21178class AgentTemplate (BaseModel ):
21279 created_at : Optional [datetime ] = None
21380
@@ -225,7 +92,7 @@ class AgentTemplate(BaseModel):
22592
22693 max_tokens : Optional [int ] = None
22794
228- model : Optional [AgentTemplateModel ] = None
95+ model : Optional [APIAgentModel ] = None
22996
23097 name : Optional [str ] = None
23198
@@ -276,7 +143,7 @@ class Agent(BaseModel):
276143 response.
277144 """
278145
279- model : Optional [AgentModel ] = None
146+ model : Optional [APIAgentModel ] = None
280147
281148 name : Optional [str ] = None
282149
0 commit comments