55from typing_extensions import Literal
66
77from .._models import BaseModel
8- from .api_model import APIModel
98from .agents .api_meta import APIMeta
109from .agents .api_links import APILinks
1110from .api_knowledge_base import APIKnowledgeBase
1817 "AgentChatbot" ,
1918 "AgentChatbotIdentifier" ,
2019 "AgentDeployment" ,
20+ "AgentModel" ,
21+ "AgentModelAgreement" ,
22+ "AgentModelVersion" ,
2123 "AgentTemplate" ,
2224 "AgentTemplateGuardrail" ,
25+ "AgentTemplateModel" ,
26+ "AgentTemplateModelAgreement" ,
27+ "AgentTemplateModelVersion" ,
2328]
2429
2530
@@ -69,12 +74,140 @@ class AgentDeployment(BaseModel):
6974 visibility : Optional [APIDeploymentVisibility ] = None
7075
7176
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+
72141class AgentTemplateGuardrail (BaseModel ):
73142 priority : Optional [int ] = None
74143
75144 uuid : Optional [str ] = None
76145
77146
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+
78211class AgentTemplate (BaseModel ):
79212 created_at : Optional [datetime ] = None
80213
@@ -92,7 +225,7 @@ class AgentTemplate(BaseModel):
92225
93226 max_tokens : Optional [int ] = None
94227
95- model : Optional [APIModel ] = None
228+ model : Optional [AgentTemplateModel ] = None
96229
97230 name : Optional [str ] = None
98231
@@ -143,7 +276,7 @@ class Agent(BaseModel):
143276 response.
144277 """
145278
146- model : Optional [APIModel ] = None
279+ model : Optional [AgentModel ] = None
147280
148281 name : Optional [str ] = None
149282
0 commit comments