File tree Expand file tree Collapse file tree 8 files changed +580
-9
lines changed
Expand file tree Collapse file tree 8 files changed +580
-9
lines changed Original file line number Diff line number Diff line change 88
99try :
1010 from vector_mcp .vectordb .postgres import PostgreSQL
11- from vector_mcp . utils import get_embedding_model
11+ from agent_utilities . embedding_utilities import create_embedding_model
1212
1313 print ("SUCCESS: Imported PostgreSQL" )
1414except ImportError as e :
1515 print (f"FAILURE: Could not import PostgreSQL: { e } " )
1616 sys .exit (1 )
1717
18- print ("\n --- Testing get_embedding_model ---" )
18+ print ("\n --- Testing create_embedding_model ---" )
1919try :
20- embed_model = get_embedding_model ()
20+ embed_model = create_embedding_model ()
2121 print (f"Got embedding model: { embed_model } " )
2222 print ("Generating test embedding..." )
2323 emb = embed_model .get_text_embedding ("test" )
Original file line number Diff line number Diff line change 1212
1313from vector_mcp .retriever .retriever import RAGRetriever
1414from vector_mcp .vectordb import ChromaVectorDB
15- from vector_mcp .vectordb .utils import (
15+ from vector_mcp .vectordb .db_utils import (
1616 optional_import_block ,
1717 require_optional_import ,
1818)
Original file line number Diff line number Diff line change 1212
1313from pydantic import Field
1414from vector_mcp .retriever .retriever import RAGRetriever
15- from vector_mcp .vectordb .utils import (
15+ from vector_mcp .vectordb .db_utils import (
1616 optional_import_block ,
1717 require_optional_import ,
1818)
Original file line number Diff line number Diff line change 77from pathlib import Path
88from typing import TYPE_CHECKING , Any
99
10- from vector_mcp .vectordb .utils import (
10+ from vector_mcp .vectordb .db_utils import (
1111 optional_import_block ,
1212 require_optional_import ,
1313)
Original file line number Diff line number Diff line change 1313from vector_mcp .retriever .retriever import RAGRetriever
1414from vector_mcp .vectordb .base import VectorDBFactory
1515from vector_mcp .vectordb .mongodb import MongoDBAtlasVectorDB
16- from vector_mcp .vectordb .utils import (
16+ from vector_mcp .vectordb .db_utils import (
1717 optional_import_block ,
1818 require_optional_import ,
1919)
Original file line number Diff line number Diff line change 1212
1313from vector_mcp .retriever .retriever import RAGRetriever
1414
15- from vector_mcp .vectordb .utils import (
15+ from vector_mcp .vectordb .db_utils import (
1616 optional_import_block ,
1717 require_optional_import ,
1818)
Original file line number Diff line number Diff line change 1212
1313from pydantic import Field
1414from vector_mcp .retriever .retriever import RAGRetriever
15- from vector_mcp .vectordb .utils import (
15+ from vector_mcp .vectordb .db_utils import (
1616 optional_import_block ,
1717 require_optional_import ,
1818)
You can’t perform that action at this time.
0 commit comments