You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
433 B
Python
26 lines
433 B
Python
from pydantic import BaseModel
|
|
from typing import Dict, Any
|
|
|
|
class ChatRequest(BaseModel):
|
|
message: str
|
|
user_id: str
|
|
|
|
class ChatResponse(BaseModel):
|
|
response: str
|
|
|
|
|
|
|
|
class ToolCreate(BaseModel):
|
|
name: str
|
|
description: str
|
|
parameters: Dict[str, Any]
|
|
|
|
|
|
class ToolResponse(BaseModel):
|
|
id: int
|
|
name: str
|
|
description: str
|
|
parameters: Dict[str, Any]
|
|
|
|
class Config:
|
|
from_attributes = True |