fix(api): fix missing custom query parameters

helixml · Nov 20, 2024 · e17b117 · e17b117
1 parent dc45bb3
commit e17b117
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/api/pkg/server/openai_chat_handlers.go b/api/pkg/server/openai_chat_handlers.go
@@ -83,6 +83,15 @@ func (s *HelixAPIServer) createChatCompletion(rw http.ResponseWriter, r *http.Re
 		AppID:       r.URL.Query().Get("app_id"),
 		AssistantID: r.URL.Query().Get("assistant_id"),
 		RAGSourceID: r.URL.Query().Get("rag_source_id"),
+		QueryParams: func() map[string]string {
+			params := make(map[string]string)
+			for key, values := range r.URL.Query() {
+				if len(values) > 0 {
+					params[key] = values[0]
+				}
+			}
+			return params
+		}(),
 	}
 
 	if user.AppID != "" {