Skip to content

Commit 022c014

Browse files
committed
chore: ObjectStore params do not relate to deepset
1 parent 551931d commit 022c014

2 files changed

Lines changed: 13 additions & 13 deletions

File tree

src/deepset_mcp/main.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -100,23 +100,23 @@ def main(
100100
typer.Option(
101101
"--object-store-backend",
102102
help="Object store backend type: 'memory' or 'redis'. "
103-
"Can also be set via DEEPSET_OBJECT_STORE_BACKEND environment variable. Default is 'memory'.",
103+
"Can also be set via OBJECT_STORE_BACKEND environment variable. Default is 'memory'.",
104104
),
105105
] = None,
106-
redis_url: Annotated[
106+
object_store_redis_url: Annotated[
107107
str | None,
108108
typer.Option(
109-
"--redis-url",
109+
"--object-store-redis-url",
110110
help="Redis connection URL (e.g., redis://localhost:6379). "
111-
"Can also be set via DEEPSET_REDIS_URL environment variable.",
111+
"Can also be set via OBJECT_STORE_REDIS_URL environment variable.",
112112
),
113113
] = None,
114114
object_store_ttl: Annotated[
115115
int,
116116
typer.Option(
117117
"--object-store-ttl",
118118
help="TTL in seconds for stored objects. Default: 600 (10 minutes). "
119-
"Can also be set via DEEPSET_OBJECT_STORE_TTL environment variable.",
119+
"Can also be set via OBJECT_STORE_TTL environment variable.",
120120
),
121121
] = 600,
122122
) -> None:
@@ -136,7 +136,7 @@ def main(
136136
:param api_key_from_auth_header: Get API key from authorization header
137137
:param transport: Type of transport to use for the MCP server
138138
:param object_store_backend: Object store backend type ('memory' or 'redis')
139-
:param redis_url: Redis connection URL (required if backend='redis')
139+
:param object_store_redis_url: Redis connection URL (required if backend='redis')
140140
:param object_store_ttl: TTL in seconds for stored objects
141141
"""
142142
# Handle --list-tools flag early
@@ -153,9 +153,9 @@ def main(
153153
docs_share_url = docs_share_url or os.getenv("DEEPSET_DOCS_SHARE_URL", DEEPSET_DOCS_DEFAULT_SHARE_URL)
154154

155155
# ObjectStore configuration
156-
backend = str(object_store_backend or os.getenv("DEEPSET_OBJECT_STORE_BACKEND", "memory"))
157-
redis_url = redis_url or os.getenv("DEEPSET_REDIS_URL")
158-
ttl = int(os.getenv("DEEPSET_OBJECT_STORE_TTL", str(object_store_ttl)))
156+
backend = str(object_store_backend or os.getenv("OBJECT_STORE_BACKEND", "memory"))
157+
redis_url = object_store_redis_url or os.getenv("OBJECT_STORE_REDIS_URL")
158+
ttl = int(os.getenv("OBJECT_STORE_TTL", str(object_store_ttl)))
159159

160160
if tools:
161161
tool_names = set(tools)
@@ -198,7 +198,7 @@ def main(
198198
deepset_docs_shareable_prototype_url=docs_share_url,
199199
get_api_key_from_authorization_header=api_key_from_auth_header,
200200
object_store_backend=backend,
201-
redis_url=redis_url,
201+
object_store_redis_url=redis_url,
202202
object_store_ttl=ttl,
203203
)
204204

src/deepset_mcp/server.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def configure_mcp_server(
2525
deepset_docs_shareable_prototype_url: str | None = None,
2626
get_api_key_from_authorization_header: bool = False,
2727
object_store_backend: str = "memory",
28-
redis_url: str | None = None,
28+
object_store_redis_url: str | None = None,
2929
object_store_ttl: int = 600,
3030
) -> None:
3131
"""Configure the MCP server with the specified tools and settings.
@@ -39,7 +39,7 @@ def configure_mcp_server(
3939
:param deepset_docs_shareable_prototype_url: Optional URL for shared prototype
4040
:param get_api_key_from_authorization_header: Whether to extract API key from authorization header
4141
:param object_store_backend: Object store backend type ('memory' or 'redis')
42-
:param redis_url: Redis connection URL (required if backend='redis')
42+
:param object_store_redis_url: Redis connection URL (required if backend='redis')
4343
:param object_store_ttl: TTL in seconds for stored objects
4444
:raises ValueError: If required parameters are missing or invalid
4545
"""
@@ -72,7 +72,7 @@ def configure_mcp_server(
7272
docs_config = None
7373

7474
# Initialize the store before registering tools
75-
store = initialize_store(backend=object_store_backend, redis_url=redis_url, ttl=object_store_ttl)
75+
store = initialize_store(backend=object_store_backend, redis_url=object_store_redis_url, ttl=object_store_ttl)
7676

7777
register_tools(
7878
mcp_server_instance=mcp_server_instance,

0 commit comments

Comments
 (0)