Support raw in OllamaService

Author: intitni

Core/Sources/CodeCompletionService/API/OllamaService.swift

@@ -233,6 +233,7 @@ extension OllamaService {
         var options: ChatCompletionRequestBody.Options
         var keep_alive: String?
         var format: String?
+        var raw: Bool?
         var suffix: String?
     }
 
@@ -248,6 +249,7 @@ extension OllamaService {
 
     func sendPrompt(
         _ prompt: String,
+        raw: Bool? = nil,
         suffix: String? = nil
     ) async throws -> ResponseStream<ChatCompletionResponseChunk> {
         let requestBody = CompletionRequestBody(
@@ -261,6 +263,7 @@ extension OllamaService {
             ),
             keep_alive: keepAlive.isEmpty ? nil : keepAlive,
             format: format == .none ? nil : format.rawValue,
+            raw: raw,
             suffix: suffix
         )
 

Core/Sources/Fundamental/PromptStrategy.swift

@@ -17,6 +17,8 @@ public protocol PromptStrategy {
     var stopWords: [String] { get }
     /// The language of the source code.
     var language: CodeLanguage? { get }
+    /// If the prompt generated is raw.
+    var promptIsRaw: Bool { get }
 
     /// Creates a prompt about the source code and relevant code snippets to be sent to the AI
     /// model.
@@ -92,5 +94,7 @@ public extension PromptStrategy {
         guard let prefix = prefix.last else { return .empty }
         return .unchanged(prefix)
     }
+    
+    var promptIsRaw: Bool { false }
 }
 

Core/Sources/SuggestionService/RequestStrategies/FillInTheMiddleRequestStrategy.swift

@@ -42,6 +42,7 @@ struct FillInTheMiddleRequestStrategy: RequestStrategy {
         var relevantCodeSnippets: [RelevantCodeSnippet] { sourceRequest.relevantCodeSnippets }
         var stopWords: [String] { ["\n\n", Tag.stop].filter { !$0.isEmpty } }
         var language: CodeLanguage? { sourceRequest.language }
+        var promptIsRaw: Bool { true }
 
         var suggestionPrefix: SuggestionPrefix {
             guard let prefix = prefix.last else { return .empty }