chandanasunkara commited on
Commit
0d2283a
·
1 Parent(s): 108073e

Embedding changes

Browse files
multi_llm_chatbot_backend/app/llm/embedding_client.py CHANGED
@@ -1,7 +1,5 @@
1
  from sentence_transformers import SentenceTransformer
2
  import os
3
- import httpx
4
- import hashlib
5
 
6
  API_KEY = os.getenv("GEMINI_API_KEY")
7
 
@@ -9,6 +7,5 @@ API_KEY = os.getenv("GEMINI_API_KEY")
9
  model = SentenceTransformer("all-MiniLM-L6-v2")
10
 
11
  def get_embedding(text: str) -> list[float]:
12
- # Use deterministic hash → convert to float vector
13
- hashed = hashlib.sha256(text.encode()).hexdigest()
14
- return [int(hashed[i:i+4], 16) / 65536.0 for i in range(0, len(hashed), 4)]
 
1
  from sentence_transformers import SentenceTransformer
2
  import os
 
 
3
 
4
  API_KEY = os.getenv("GEMINI_API_KEY")
5
 
 
7
  model = SentenceTransformer("all-MiniLM-L6-v2")
8
 
9
  def get_embedding(text: str) -> list[float]:
10
+ embedding = model.encode(text, convert_to_numpy=True)
11
+ return embedding.tolist()
 
multi_llm_chatbot_backend/app/main.py CHANGED
@@ -1,7 +1,7 @@
1
  # app/main.py
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
- from app.api.routes import router as api_router
5
  from dotenv import load_dotenv
6
  import os
7
 
@@ -24,7 +24,7 @@ app.add_middleware(
24
 
25
  # Include route definitions
26
  # app.include_router(router)
27
- app.include_router(api_router, prefix="/api")
28
 
29
  @app.get("/")
30
  def root():
 
1
  # app/main.py
2
  from fastapi import FastAPI
3
  from fastapi.middleware.cors import CORSMiddleware
4
+ from app.api.routes import router
5
  from dotenv import load_dotenv
6
  import os
7
 
 
24
 
25
  # Include route definitions
26
  # app.include_router(router)
27
+ app.include_router(router)
28
 
29
  @app.get("/")
30
  def root():