diff --git a/modelsAPI/model_api.py b/modelsAPI/model_api.py index e55e56b..4473b6f 100644 --- a/modelsAPI/model_api.py +++ b/modelsAPI/model_api.py @@ -5,7 +5,6 @@ import os import numpy as np from openai import OpenAI from config import LLM_CONFIG, EMBEDDING_CONFIG -from neo4j_graphrag.embeddings.base import Embedder from openai import AsyncOpenAI @@ -353,24 +352,7 @@ class OpenaiAPI: return 0.5 * jaccard + 0.5 * len_penalty -class LocalBgeM3Embeddings(Embedder): - def __init__(self, base_url: str, api_key: str, model_name: str = "bge-m3"): - self.base_url = base_url - self.api_key = api_key - self.model_name = model_name - def embed_query(self, text: str) -> List[float]: - return self.embed_documents([text])[0] - - def embed_documents(self, texts: List[str]) -> List[List[float]]: - url = self.base_url - with httpx.Client(timeout=60) as client: - response = client.post( - url, - json={"model": self.model_name, "input": texts}, - headers={"Authorization": f"Bearer {self.api_key}"}, - ) - return [item["embedding"] for item in response.json()["data"]] if __name__ == "__main__":