File size: 629 Bytes
56a63a1
 
 
 
 
 
e0cbbfe
56a63a1
 
e0cbbfe
56a63a1
 
 
 
 
 
e0cbbfe
56a63a1
e0cbbfe
56a63a1
 
e0cbbfe
56a63a1
e0cbbfe
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
from langchain.schema import HumanMessage, AIMessage  # type: ignore
from .file import File


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

    def __str__(self):
        return self.content

    def __repr__(self):
        return f"UserRequest(content={self.content}, files={self.files})"


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

    def __str__(self):
        return self.content

    def __repr__(self):
        return f"CodeInterpreterResponse(content={self.content}, files={self.files})"