Update proxy_plugin.py

Author: codelion

optillm/plugins/proxy_plugin.py

@@ -120,12 +120,17 @@ def run(system_prompt: str, initial_query: str, client, model: str,
 
         if not config.get('providers'):
             logger.warning("No providers configured, falling back to original client")
+            # Strip stream parameter to force complete response
+            api_config = dict(request_config or {})
+            api_config.pop('stream', None)
+
             response = client.chat.completions.create(
                 model=model,
                 messages=[
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": initial_query}
-                ]
+                ],
+                **api_config
             )
             # Return full response dict to preserve all usage information
             response_dict = response.model_dump() if hasattr(response, 'model_dump') else response
@@ -223,12 +228,17 @@ def run(system_prompt: str, initial_query: str, client, model: str,
         logger.error(f"Proxy plugin error: {e}", exc_info=True)
         # Fallback to original client
         logger.info("Falling back to original client")
+        # Strip stream parameter to force complete response
+        api_config = dict(request_config or {})
+        api_config.pop('stream', None)
+
         response = client.chat.completions.create(
             model=model,
             messages=[
                 {"role": "system", "content": system_prompt},
                 {"role": "user", "content": initial_query}
-            ]
+            ],
+            **api_config
         )
         # Return full response dict to preserve all usage information
         response_dict = response.model_dump() if hasattr(response, 'model_dump') else response