File size: 645 Bytes
56a63a1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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})"