Vela commited on
Commit
935efff
·
1 Parent(s): 172e21d

modified gemini model service

Browse files
application/services/gemini_model.py CHANGED
@@ -8,7 +8,7 @@ from application.utils import logger
8
 
9
  logger=logger.get_logger()
10
 
11
- client = genai.Client(api_key=os.getenv("gemini_api_key"))
12
 
13
  # PROMPT = (
14
  # """You are a PDF parsing agent. Your job is to extract GHG Protocol Parameters
 
8
 
9
  logger=logger.get_logger()
10
 
11
+ client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
12
 
13
  # PROMPT = (
14
  # """You are a PDF parsing agent. Your job is to extract GHG Protocol Parameters
application/services/llm_service.py CHANGED
@@ -8,7 +8,7 @@ from application.utils import logger
8
  from application.schemas.response_schema import RESPONSE_FORMAT,GEMINI_RESPONSE_FORMAT
9
 
10
  logger = logger.get_logger()
11
- client = OpenAI(api_key=os.getenv("openai_api_key"))
12
 
13
  # --- Constants ---
14
 
 
8
  from application.schemas.response_schema import RESPONSE_FORMAT,GEMINI_RESPONSE_FORMAT
9
 
10
  logger = logger.get_logger()
11
+ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
12
 
13
  # --- Constants ---
14
 
application/services/supabase_service.py DELETED
@@ -1,20 +0,0 @@
1
- import json
2
- import os
3
- from datetime import datetime
4
- from supabase import create_client, StorageException
5
- from utils import logger
6
- from dotenv import load_dotenv
7
-
8
- # Logger Initialization
9
- logger = logger.get_logger()
10
-
11
- # Load Environment Variables
12
- load_dotenv()
13
- SUPABASE_URL = os.getenv('SUPABASE_URL')
14
- SUPABASE_KEY = os.getenv('SUPABASE_KEY')
15
- SUPABASE_BUCKET = os.getenv('SUPABASE_BUCKET')
16
- LLM_MODEL_NAME = os.getenv('LLM_MODEL_NAME')
17
- BUCKET_FOLDER = "chat-history"
18
-
19
- # Supabase Client Initialization
20
- supabase = create_client(SUPABASE_URL, SUPABASE_KEY)