24 lines
718 B
Python
24 lines
718 B
Python
|
from codeinterpreterapi import CodeInterpreterSession
|
||
|
|
from codeinterpreterapi import File
|
||
|
|
|
||
|
|
|
||
|
|
class CodeInterpreter:
|
||
|
|
def __init__(self, db_manager):
|
||
|
|
self.db_manager = db_manager
|
||
|
|
|
||
|
|
async def process(self, prompt, uploaded_files):
|
||
|
|
files = []
|
||
|
|
for uploaded_file in uploaded_files:
|
||
|
|
file = File(
|
||
|
|
name=uploaded_file.name,
|
||
|
|
content=uploaded_file.read()
|
||
|
|
)
|
||
|
|
files.append(file)
|
||
|
|
async with CodeInterpreterSession(model='gpt-4') as session:
|
||
|
|
response = await session.generate_response(
|
||
|
|
prompt,
|
||
|
|
files=files,
|
||
|
|
detailed_error=True
|
||
|
|
)
|
||
|
|
return response
|