Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bug with using enum as string

Browse Source
This commit is contained in:
Abi Raja 2024-03-19 10:30:58 -04:00
parent 81c4fbe28d
commit 212aa228ab
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
backend/llm.py
View File

@ -35,7 +35,12 @@ async def stream_openai_response(
model = Llm.GPT_4_VISION model = Llm.GPT_4_VISION
# Base parameters # Base parameters
params = {"model": model, "messages": messages, "stream": True, "timeout": 600} params = {
"model": model.value,
"messages": messages,
"stream": True,
"timeout": 600,
}
# Add 'max_tokens' only if the model is a GPT4 vision model # Add 'max_tokens' only if the model is a GPT4 vision model
if model == Llm.GPT_4_VISION: if model == Llm.GPT_4_VISION: