Michael Poluektov 9 months ago
parent
commit
ed205d82e8
2 changed files with 3 additions and 3 deletions
  1. 2 2
      backend/apps/ollama/main.py
  2. 1 1
      backend/apps/openai/main.py

+ 2 - 2
backend/apps/ollama/main.py

@@ -729,7 +729,7 @@ async def generate_chat_completion(
     payload = {
     payload = {
         **form_data.model_dump(exclude_none=True, exclude=["metadata"]),
         **form_data.model_dump(exclude_none=True, exclude=["metadata"]),
     }
     }
-    payload.pop("metadata")
+    payload.pop("metadata", None)
 
 
     model_id = form_data.model
     model_id = form_data.model
     model_info = Models.get_model_by_id(model_id)
     model_info = Models.get_model_by_id(model_id)
@@ -788,7 +788,7 @@ async def generate_openai_chat_completion(
 ):
 ):
     completion_form = OpenAIChatCompletionForm(**form_data)
     completion_form = OpenAIChatCompletionForm(**form_data)
     payload = {**completion_form.model_dump(exclude_none=True, exclude=["metadata"])}
     payload = {**completion_form.model_dump(exclude_none=True, exclude=["metadata"])}
-    payload.pop("metadata")
+    payload.pop("metadata", None)
 
 
     model_id = completion_form.model
     model_id = completion_form.model
     model_info = Models.get_model_by_id(model_id)
     model_info = Models.get_model_by_id(model_id)

+ 1 - 1
backend/apps/openai/main.py

@@ -359,7 +359,7 @@ async def generate_chat_completion(
 ):
 ):
     idx = 0
     idx = 0
     payload = {**form_data}
     payload = {**form_data}
-    payload.pop("metadata")
+    payload.pop("metadata", None)
 
 
     model_id = form_data.get("model")
     model_id = form_data.get("model")
     model_info = Models.get_model_by_id(model_id)
     model_info = Models.get_model_by_id(model_id)