From cd70022d67265aa3be579e7f4e22cd878e39d681 Mon Sep 17 00:00:00 2001
From: Defeng <dfhu921@163.com>
Date: Fri, 17 Jul 2026 15:19:55 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96models=5Fapi?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modelsAPI/model_api.py | 18 ------------------
 1 file changed, 18 deletions(-)

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__":