File size: 460 Bytes
cd99429
 
e1776b1
 
cd99429
 
 
 
 
 
e1776b1
cd99429
e1776b1
 
cd99429
 
e1776b1
 
cd99429
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from langchain.schema import HumanMessage, AIMessage  # type: ignore
from pydantic import BaseModel


class File(BaseModel):
    name: str
    content: bytes


class UserRequest(HumanMessage):
    text: str = ""
    files: list[File] = []


class CodeInterpreterResponse(AIMessage):
    files: list[File] = []
    final_code: str = ""
    final_output: str = ""


class CodeInput(BaseModel): 
    code: str
    

class FileInput(BaseModel): 
    filename: str