fix ollama for cline

Author: yrobla

src/codegate/providers/completion/base.py

@@ -20,7 +20,7 @@ async def execute_completion(
         api_key: Optional[str],
         stream: bool = False,  # TODO: remove this param?
         is_fim_request: bool = False,
-        is_cline_client: bool = False,
+        base_tool: Optional[str] = "",
     ) -> Union[ModelResponse, AsyncIterator[ModelResponse]]:
         """Execute the completion request"""
         pass

src/codegate/providers/ollama/completion_handler.py

@@ -63,15 +63,13 @@ async def ollama_stream_generator(
                         if field in chunk_dict:
                             response[field] = chunk_dict[field]
 
-                    print("in cline")
-                    print(json.dumps(response))
-                    yield f"data: {json.dumps(response)}\n"
+                    yield f"\ndata: {json.dumps(response)}\n"
             except Exception as e:
                 logger.error(f"Error in stream generator: {str(e)}")
-                yield f"data: {json.dumps({'error': str(e), 'type': 'error', 'choices': []})}\n"
+                yield f"\ndata: {json.dumps({'error': str(e), 'type': 'error', 'choices': []})}\n"
     except Exception as e:
         logger.error(f"Stream error: {str(e)}")
-        yield f"data: {json.dumps({'error': str(e), 'type': 'error', 'choices': []})}\n"
+        yield f"\ndata: {json.dumps({'error': str(e), 'type': 'error', 'choices': []})}\n"
 
 
 class OllamaShim(BaseCompletionHandler):
@@ -115,7 +113,9 @@ def _create_streaming_response(self, stream: AsyncIterator[ChatResponse]) -> Str
             headers={
                 "Cache-Control": "no-cache",
                 "Connection": "keep-alive",
+                "Transfer-Encoding": "chunked",
             },
+            status_code=200,
         )
 
     def _create_json_response(