From 62758f0fcd680f321739f99b4664dd101a7b643d Mon Sep 17 00:00:00 2001 From: papayalabs Date: Tue, 26 Nov 2024 19:22:19 +0100 Subject: [PATCH] Refactor --- app/services/ai_backend.rb | 3 +-- app/services/ai_backend/gemini.rb | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/app/services/ai_backend.rb b/app/services/ai_backend.rb index 1f3753f3..d4204f08 100644 --- 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 index 7ae77ec4..c04aced8 100644 --- 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