Hackathons/Hoya26
1
0
Fork 0
mirror of https://github.com/SirBlobby/Hoya26.git synced 2026-02-04 03:34:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Code Update

Browse Source
This commit is contained in:
default
2026-01-24 04:01:05 +00:00
parent c87279ed52
commit aad7e6e08d
13 changed files with 18 additions and 469 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
backend/src/rag/embeddings.py
View File

@@ -2,9 +2,6 @@ from google import genai
import os
def get_embedding(text, model="gemini-embedding-001"):
"""
Generates an embedding for the given text using the Gemini API.
"""
api_key = os.environ.get("GOOGLE_API_KEY")
if not api_key:
raise ValueError("GOOGLE_API_KEY environment variable not set")
@@ -17,9 +14,6 @@ def get_embedding(text, model="gemini-embedding-001"):
return result.embeddings[0].values
def get_embeddings_batch(texts, model="gemini-embedding-001"):
"""
Generates embeddings for a list of texts.
"""
api_key = os.environ.get("GOOGLE_API_KEY")
if not api_key:
raise ValueError("GOOGLE_API_KEY environment variable not set")
@@ -29,5 +23,4 @@ def get_embeddings_batch(texts, model="gemini-embedding-001"):
model=model,
contents=texts
)
# The SDK returns a list of embedding objects
return [emb.values for emb in result.embeddings]