Refactor

AllYourBot · Nov 26, 2024 · 62758f0 · 62758f0
1 parent fc2e349
commit 62758f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/app/services/ai_backend.rb b/app/services/ai_backend.rb
@@ -21,8 +21,7 @@ def get_oneoff_message(instructions, messages, params = {})
     response = @client.send(client_method_name, ** @client_config)
 
     response.dig("content", 0, "text") ||
-      response.dig("choices", 0, "message", "content") ||
-        response.dig("candidates",0,"content","parts",0,"text")
+      response.dig("choices", 0, "message", "content")
   end
 
   def stream_next_conversation_message(&chunk_handler)

diff --git a/app/services/ai_backend/gemini.rb b/app/services/ai_backend/gemini.rb
@@ -48,7 +48,7 @@ def set_client_config(config)
   end
 
   def get_oneoff_message(instructions, messages, params = {})
-    response = @client.generate_content( { system_instruction: { role: "user", parts: { text: instructions }}, contents: { role: "user", parts: { text: messages.first }}})
+    response = @client.generate_content( { system_instruction: system_message(instructions), contents: { role: "user", parts: { text: messages.first }}})
     response.dig("candidates",0,"content","parts",0,"text")
   end