pre-commit fixes

llama_stack/providers/remote/vector_io/pgvector/pgvector.py

@@ -58,7 +58,11 @@ class PGVectorIndex(EmbeddingIndex):
     def __init__(self, vector_db: VectorDB, dimension: int, conn):
         self.conn = conn
         with conn.cursor(cursor_factory=psycopg2.extras.DictCursor) as cur:
-            self.table_name = f"vector_store_{vector_db.identifier}"
+            # Sanitize the table name by replacing hyphens with underscores
+            # SQL doesn't allow hyphens in table names, and vector_db.identifier may contain hyphens
+            # when created with patterns like "test-vector-db-{uuid4()}"
+            sanitized_identifier = vector_db.identifier.replace("-", "_")
+            self.table_name = f"vector_store_{sanitized_identifier}"
 
             cur.execute(
                 f"""

llama_stack/providers/remote/vector_io/qdrant/config.py

@@ -4,7 +4,7 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
-from typing import Optional
+from typing import Any, Dict, Optional
 
 from pydantic import BaseModel
 
@@ -24,3 +24,9 @@ class QdrantVectorIOConfig(BaseModel):
     timeout: Optional[int] = None
     host: Optional[str] = None
     path: Optional[str] = None
+
+    @classmethod
+    def sample_run_config(cls, **kwargs: Any) -> Dict[str, Any]:
+        return {
+            "api_key": "${env.QDRANT_API_KEY}",
+        }

llama_stack/providers/remote/vector_io/sample/__init__.py

@@ -1,17 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-#
-# This source code is licensed under the terms described in the LICENSE file in
-# the root directory of this source tree.
-
-from typing import Any
-
-from .config import SampleVectorIOConfig
-
-
-async def get_adapter_impl(config: SampleVectorIOConfig, _deps) -> Any:
-    from .sample import SampleVectorIOImpl
-
-    impl = SampleVectorIOImpl(config)
-    await impl.initialize()
-    return impl

llama_stack/providers/remote/vector_io/sample/config.py

@@ -1,12 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-#
-# This source code is licensed under the terms described in the LICENSE file in
-# the root directory of this source tree.
-
-from pydantic import BaseModel
-
-
-class SampleVectorIOConfig(BaseModel):
-    host: str = "localhost"
-    port: int = 9999

llama_stack/providers/remote/vector_io/sample/sample.py

@@ -1,26 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-#
-# This source code is licensed under the terms described in the LICENSE file in
-# the root directory of this source tree.
-
-from llama_stack.apis.vector_dbs import VectorDB
-from llama_stack.apis.vector_io import VectorIO
-
-from .config import SampleVectorIOConfig
-
-
-class SampleVectorIOImpl(VectorIO):
-    def __init__(self, config: SampleVectorIOConfig):
-        self.config = config
-
-    async def register_vector_db(self, vector_db: VectorDB) -> None:
-        # these are the vector dbs the Llama Stack will use to route requests to this provider
-        # perform validation here if necessary
-        pass
-
-    async def initialize(self):
-        pass
-
-    async def shutdown(self):
-        pass

llama_stack/providers/remote/vector_io/weaviate/config.py

@@ -4,6 +4,8 @@
 # This source code is licensed under the terms described in the LICENSE file in
 # the root directory of this source tree.
 
+from typing import Any, Dict
+
 from pydantic import BaseModel
 
 
@@ -13,4 +15,6 @@ class WeaviateRequestProviderData(BaseModel):
 
 
 class WeaviateVectorIOConfig(BaseModel):
-    pass
+    @classmethod
+    def sample_run_config(cls, **kwargs: Any) -> Dict[str, Any]:
+        return {}