bug fixes potentially

models/gemini.py

@@ -1,15 +1,25 @@
+import os
 from typing import Optional
 import sys
 
+import vertexai
+from dotenv import load_dotenv
+
 sys.path.append("../")
 from setup.vertex_ai_setup import initialize_vertexai_params
 from vertexai.generative_models import GenerativeModel
 
+load_dotenv()
+VERTEXAI_PROJECT = os.environ["VERTEXAI_PROJECT"]
+
 
-def get_gemini_response(prompt_text, model, parameters: Optional = None) -> str:
+def get_gemini_response(prompt_text, model, parameters: Optional = None, location: Optional[str] = "us-central1") -> str:
     initialize_vertexai_params()
+
     if model is None or parameters is None:
         model = "gemini-1.0-pro"
     model = GenerativeModel(model)
+
     model_response = model.generate_content(prompt_text)
+
     return model_response.text

setup/vertex_ai_setup.py

@@ -7,23 +7,27 @@ import json
 import base64
 
 load_dotenv()
-
-# TODO: fix it in spaces
-
 VERTEXAI_PROJECT = os.environ["VERTEXAI_PROJECT"]
 
 
 def decode_service_key():
     encoded_key = os.environ["GOOGLE_APPLICATION_CREDENTIALS"]
     original_service_key = json.loads(base64.b64decode(encoded_key).decode('utf-8'))
-    return original_service_key
+    if original_service_key:
+        return original_service_key
+    return None
 
 
 def initialize_vertexai_params(location: Optional[str] = "us-central1"):
-    GOOGLE_APPLICATION_CREDENTIALS = decode_service_key()
-    service_account.Credentials.from_service_account_info(GOOGLE_APPLICATION_CREDENTIALS, scopes=["https://www"
-                                                                                                  ".googleapis.com/auth/cloud-platform"])
+    credentials = decode_service_key()
+    creds_file_name = os.getcwd() + "/.config/application_default_credentials.json"
+    with open(creds_file_name, 'w', encoding='utf-8') as file:
+        json.dump(credentials, file, ensure_ascii=False, indent=4)
+
+    os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = creds_file_name
 
-    print("service account worked")
+    service_account.Credentials.from_service_account_file(
+        filename=os.environ["GOOGLE_APPLICATION_CREDENTIALS"],
+        scopes=["https://www.googleapis.com/auth/cloud-platform"],
+    )
     vertexai.init(project=VERTEXAI_PROJECT, location=location)
-    print("init worked")