fix

README.md

@@ -46,9 +46,7 @@ Get Involved: [Discuss and contribute on GitHub](https://github.com/yanolja/aren
    Set your environment variables and run the app:
 
    ```shell
-   GOOGLE_CLOUD_PROJECT=<your project id> \
    CREDENTIALS_PATH=<your crednetials path> \
-   MODELS_SECRET=<your secret> \
    OPENAI_API_KEY=<your key> \
    ANTHROPIC_API_KEY=<your key> \
    MISTRAL_API_KEY=<your key> \

model.py

@@ -2,29 +2,11 @@
 This module contains functions to interact with the models.
 """
 
-from enum import Enum
 import json
-import os
 from typing import List
 
-from google.cloud import secretmanager
-from google.oauth2 import service_account
 import litellm
 
-from credentials import get_credentials_json
-
-GOOGLE_CLOUD_PROJECT = os.environ.get("GOOGLE_CLOUD_PROJECT")
-MODELS_SECRET = os.environ.get("MODELS_SECRET")
-
-secretmanager_client = secretmanager.SecretManagerServiceClient(
-    credentials=service_account.Credentials.from_service_account_info(
-        get_credentials_json()))
-models_secret = secretmanager_client.access_secret_version(
-    name=secretmanager_client.secret_version_path(GOOGLE_CLOUD_PROJECT,
-                                                  MODELS_SECRET, "latest"))
-decoded_secret = models_secret.payload.data.decode("UTF-8")
-
-
 DEFAULT_SUMMARIZE_INSTRUCTION = "Summarize the given text without changing the language of it."  # pylint: disable=line-too-long
 DEFAULT_TRANSLATE_INSTRUCTION = "Translate the given text from {source_lang} to {target_lang}."  # pylint: disable=line-too-long
 

response.py

@@ -77,13 +77,7 @@ def get_responses(prompt: str, category: str, source_lang: str,
     instruction = get_instruction(category, model, source_lang, target_lang)
     try:
       # TODO(#1): Allow user to set configuration.
-      response = model.completion(messages=[{
-          "role": "system",
-          "content": instruction
-      }, {
-          "role": "user",
-          "content": prompt
-      }])
+      response = model.completion(instruction, prompt)
       create_history(category, model.name, instruction, prompt, response)
       responses.append(response)