Skip to content

Commit f340145

Browse files
committed
Bump version: 1.1.17 → 1.1.18
1 parent 5167bbb commit f340145

File tree

6 files changed

+52
-6
lines changed

6 files changed

+52
-6
lines changed

.bumpversion.cfg

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
[bumpversion]
2-
current_version = 1.1.17
2+
current_version = 1.1.18
33
commit = True
44
tag = True
55

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,6 @@ ENV HOST=${HOST} \
5454
RUN apt-get update \
5555
&& apt-get install -y curl nano libpq-dev \
5656
&& curl -LsSf https://astral.sh/uv/install.sh | sh \
57-
&& uv pip install --system --upgrade --verbose --no-cache --break-system-packages --prerelease=allow vector-mcp[postgres,chromadb,couchbase,qdrant,mongodb,a2a]>=1.1.17
57+
&& uv pip install --system --upgrade --verbose --no-cache --break-system-packages --prerelease=allow vector-mcp[postgres,chromadb,couchbase,qdrant,mongodb,a2a]>=1.1.18
5858

5959
CMD ["vector-mcp"]

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
![PyPI - Wheel](https://img.shields.io/pypi/wheel/vector-mcp)
2222
![PyPI - Implementation](https://img.shields.io/pypi/implementation/vector-mcp)
2323

24-
*Version: 1.1.17*
24+
*Version: 1.1.18*
2525

2626
## Overview
2727

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "vector-mcp"
7-
version = "1.1.17"
7+
version = "1.1.18"
88
description = "Integrate RAG into AI Agents via MCP Server. Supports multiple Vector database technologies."
99
readme = "README.md"
1010
authors = [{ name = "Audel Rouhi", email = "knucklessg1@gmail.com" }]

vector_mcp/vector_agent.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ class AgentState(BaseModel):
4141
from pydantic_ai.ui import SSE_CONTENT_TYPE
4242
from pydantic_ai.ui.ag_ui import AGUIAdapter
4343

44-
__version__ = "1.1.17"
44+
__version__ = "1.1.18"
4545

4646
logging.basicConfig(
4747
level=logging.INFO,
@@ -89,6 +89,12 @@ class AgentState(BaseModel):
8989
"CRITICAL RULE: \n"
9090
"1. List all available collections using`list_collections` to discover which knowledge base collections exist.\n"
9191
"2. Select the most relevant collection (or collections) from the list.\n"
92+
" - `decisions`: For storing key decisions made.\n"
93+
" - `user`: For storing user preferences and information.\n"
94+
" - `myself`: For storing agent's self-reflection, identity, or learnings.\n"
95+
" - `knowledge`: For general knowledge base.\n"
96+
" - `tasks`: For tracking tasks and project status.\n"
97+
" - `patterns`: For reusable patterns and coding standards.\n"
9298
"- If no specific collection is mentioned in the question, default to the memory collection.\n"
9399
'- If the question implies a domain (e.g., "ag-ui", "pydantic", "debug"), infer the most relevant collection from the context.\n'
94100
"3. Perform a hybrid search using the `search` tool with the selected collection(s) to retrieve relevant information.\n"

vector_mcp/vector_mcp.py

Lines changed: 41 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
from vector_mcp.utils import to_integer, to_boolean
2727
from vector_mcp.middlewares import UserTokenMiddleware, JWTClaimsLoggingMiddleware
2828

29-
__version__ = "1.1.17"
29+
__version__ = "1.1.18"
3030

3131
logging.basicConfig(
3232
level=logging.DEBUG, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
@@ -52,6 +52,7 @@
5252
DEFAULT_TRANSPORT = os.environ.get("TRANSPORT", "stdio")
5353
DEFAULT_HOST = os.environ.get("HOST", "0.0.0.0")
5454
DEFAULT_PORT = to_integer(os.environ.get("PORT", "8000"))
55+
DEFAULT_SSL_VERIFY = to_boolean(os.environ.get("SSL_VERIFY", "False"))
5556
DEFAULT_DB_HOST = os.environ.get("DB_HOST", None)
5657
DEFAULT_DB_PORT = os.environ.get("DB_PORT", None)
5758
DEFAULT_DATABASE_TYPE = os.environ.get("DATABASE_TYPE", "chromadb").lower()
@@ -77,6 +78,43 @@
7778
logger.info(f"Global chunk size set to: {chunk_size}")
7879

7980

81+
DEFAULT_COLLECTIONS = [
82+
"decisions",
83+
"user",
84+
"myself",
85+
"knowledge",
86+
"tasks",
87+
"patterns",
88+
]
89+
90+
91+
def create_default_collections(
92+
db_type: str = DEFAULT_DATABASE_TYPE,
93+
db_path: str = DEFAULT_DATABASE_PATH,
94+
host: Optional[str] = DEFAULT_DB_HOST,
95+
port: Optional[str] = DEFAULT_DB_PORT,
96+
db_name: Optional[str] = DEFAULT_DBNAME,
97+
username: Optional[str] = DEFAULT_USERNAME,
98+
password: Optional[str] = DEFAULT_PASSWORD,
99+
):
100+
for collection in DEFAULT_COLLECTIONS:
101+
try:
102+
initialize_retriever(
103+
db_type=db_type,
104+
db_path=db_path,
105+
host=host,
106+
port=port,
107+
db_name=db_name,
108+
username=username,
109+
password=password,
110+
collection_name=collection,
111+
ensure_collection_exists=True,
112+
)
113+
logger.info(f"Ensured default collection exists: {collection}")
114+
except Exception as e:
115+
logger.error(f"Failed to create default collection {collection}: {e}")
116+
117+
80118
def initialize_retriever(
81119
db_type: str = DEFAULT_DATABASE_TYPE,
82120
db_path: str = DEFAULT_DATABASE_PATH,
@@ -1453,6 +1491,8 @@ def vector_mcp():
14531491
for mw in middlewares:
14541492
mcp.add_middleware(mw)
14551493

1494+
create_default_collections()
1495+
14561496
print("\nStarting Vector MCP Server")
14571497
print(f" Transport: {args.transport.upper()}")
14581498
print(f" Auth: {args.auth_type}")

0 commit comments

Comments
 (0)