Spaces:
Sleeping
Sleeping
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("
|
| 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("
|
| 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|