phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-07 02:58:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

removing print statements and reverting some changing in routing tables

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-02-10 17:15:43 -05:00
parent 75a480267a
commit 99a83f17de
2 changed files with 1 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/distribution/routers/routing_tables.py
View file

@ -185,9 +185,7 @@ class CommonRoutingTableImpl(RoutingTable):
obj.provider_id = list(self.impls_by_provider_id.keys())[0] obj.provider_id = list(self.impls_by_provider_id.keys())[0]
if obj.provider_id not in self.impls_by_provider_id: if obj.provider_id not in self.impls_by_provider_id:
raise ValueError( raise ValueError(f"Provider `{obj.provider_id}` not found")
f"Provider `{obj.provider_id}` not found \navailable providers: {self.impls_by_provider_id.keys()}"
)
p = self.impls_by_provider_id[obj.provider_id] p = self.impls_by_provider_id[obj.provider_id]
@ -337,7 +335,6 @@ class VectorDBsRoutingTable(CommonRoutingTableImpl, VectorDBs):
"embedding_model": embedding_model, "embedding_model": embedding_model,
"embedding_dimension": model.metadata["embedding_dimension"], "embedding_dimension": model.metadata["embedding_dimension"],
} }
print(f"Registering vector db {vector_db_data} with embedding model {embedding_model}")
vector_db = TypeAdapter(VectorDB).validate_python(vector_db_data) vector_db = TypeAdapter(VectorDB).validate_python(vector_db_data)
await self.register_object(vector_db) await self.register_object(vector_db)
return vector_db return vector_db

2
llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py
View file

@ -49,7 +49,6 @@ class SQLiteVecIndex(EmbeddingIndex):
async def initialize(self) -> None: async def initialize(self) -> None:
cur = self.connection.cursor() cur = self.connection.cursor()
print(f"Creating tables {self.metadata_table} and {self.vector_table}")
# Create the table to store chunk metadata. # Create the table to store chunk metadata.
cur.execute(f""" cur.execute(f"""
CREATE TABLE IF NOT EXISTS {self.metadata_table} ( CREATE TABLE IF NOT EXISTS {self.metadata_table} (
@ -135,7 +134,6 @@ class SQLiteVecVectorIOImpl(VectorIO, VectorDBsProtocolPrivate):
async def initialize(self) -> None: async def initialize(self) -> None:
# Open a connection to the SQLite database (the file is specified in the config). # Open a connection to the SQLite database (the file is specified in the config).
print(f"Connecting to SQLite database at {self.config.db_path}")
self.connection = sqlite3.connect(self.config.db_path) self.connection = sqlite3.connect(self.config.db_path)
self.connection.enable_load_extension(True) self.connection.enable_load_extension(True)
sqlite_vec.load(self.connection) sqlite_vec.load(self.connection)