aboutsummaryrefslogtreecommitdiffstats
path: root/useragents/src/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'useragents/src/models.py')
-rw-r--r--useragents/src/models.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/useragents/src/models.py b/useragents/src/models.py
new file mode 100644
index 0000000..14d37fc
--- /dev/null
+++ b/useragents/src/models.py
@@ -0,0 +1,12 @@
+import pydantic
+
+
+class AgentRequest(pydantic.BaseModel):
+ agent_name: str
+ instructions: str
+ query: str
+
+
+class AgentResponse(pydantic.BaseModel):
+ agent_name: str
+ response: str