diff --git a/pyproject.toml b/pyproject.toml index eea515b09..3e16dc08f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -356,6 +356,10 @@ exclude = [ module = [ "yaml", "fire", + "redis.asyncio", + "psycopg2", + "psycopg2.extras", + "psycopg2.extensions", "torchtune.*", "fairscale.*", "torchvision.*", diff --git a/src/llama_stack/core/storage/kvstore/postgres/postgres.py b/src/llama_stack/core/storage/kvstore/postgres/postgres.py index 029b902ec..39c3fd2e2 100644 --- a/src/llama_stack/core/storage/kvstore/postgres/postgres.py +++ b/src/llama_stack/core/storage/kvstore/postgres/postgres.py @@ -6,9 +6,9 @@ from datetime import datetime -import psycopg2 -from psycopg2.extensions import connection as PGConnection -from psycopg2.extras import DictCursor +import psycopg2 # type: ignore[import-not-found] +from psycopg2.extensions import connection as PGConnection # type: ignore[import-not-found] +from psycopg2.extras import DictCursor # type: ignore[import-not-found] from llama_stack.log import get_logger from llama_stack_api.internal.kvstore import KVStore diff --git a/src/llama_stack/core/storage/kvstore/redis/redis.py b/src/llama_stack/core/storage/kvstore/redis/redis.py index ff6803ce2..2b35a22e1 100644 --- a/src/llama_stack/core/storage/kvstore/redis/redis.py +++ b/src/llama_stack/core/storage/kvstore/redis/redis.py @@ -6,7 +6,7 @@ from datetime import datetime -from redis.asyncio import Redis +from redis.asyncio import Redis # type: ignore[import-not-found] from llama_stack_api.internal.kvstore import KVStore diff --git a/src/llama_stack/core/storage/kvstore/sqlite/config.py b/src/llama_stack/core/storage/kvstore/sqlite/config.py deleted file mode 100644 index 0f8fa0a95..000000000 --- a/src/llama_stack/core/storage/kvstore/sqlite/config.py +++ /dev/null @@ -1,20 +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, Field - -from llama_stack_api import json_schema_type - - -@json_schema_type -class SqliteControlPlaneConfig(BaseModel): - db_path: str = Field( - description="File path for the sqlite database", - ) - table_name: str = Field( - default="llamastack_control_plane", - description="Table into which all the keys will be placed", - )