agent-brain-rag 1.1.0__py3-none-any.whl → 1.2.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {agent_brain_rag-1.1.0.dist-info → agent_brain_rag-1.2.0.dist-info}/METADATA +17 -14
- agent_brain_rag-1.2.0.dist-info/RECORD +31 -0
- agent_brain_rag-1.2.0.dist-info/entry_points.txt +4 -0
- {doc_serve_server → agent_brain_server}/__init__.py +1 -1
- {doc_serve_server → agent_brain_server}/api/main.py +62 -26
- {doc_serve_server → agent_brain_server}/api/routers/health.py +3 -2
- {doc_serve_server → agent_brain_server}/api/routers/index.py +1 -1
- {doc_serve_server → agent_brain_server}/api/routers/query.py +3 -3
- {doc_serve_server → agent_brain_server}/indexing/__init__.py +4 -4
- {doc_serve_server → agent_brain_server}/indexing/bm25_index.py +1 -1
- {doc_serve_server → agent_brain_server}/indexing/chunking.py +1 -1
- {doc_serve_server → agent_brain_server}/indexing/embedding.py +1 -1
- {doc_serve_server → agent_brain_server}/models/health.py +2 -2
- {doc_serve_server → agent_brain_server}/services/indexing_service.py +4 -4
- {doc_serve_server → agent_brain_server}/services/query_service.py +9 -4
- {doc_serve_server → agent_brain_server}/storage/vector_store.py +1 -1
- agent_brain_rag-1.1.0.dist-info/RECORD +0 -31
- agent_brain_rag-1.1.0.dist-info/entry_points.txt +0 -3
- {agent_brain_rag-1.1.0.dist-info → agent_brain_rag-1.2.0.dist-info}/WHEEL +0 -0
- {doc_serve_server → agent_brain_server}/api/__init__.py +0 -0
- {doc_serve_server → agent_brain_server}/api/routers/__init__.py +0 -0
- {doc_serve_server → agent_brain_server}/config/__init__.py +0 -0
- {doc_serve_server → agent_brain_server}/config/settings.py +0 -0
- {doc_serve_server → agent_brain_server}/indexing/document_loader.py +0 -0
- {doc_serve_server → agent_brain_server}/locking.py +0 -0
- {doc_serve_server → agent_brain_server}/models/__init__.py +0 -0
- {doc_serve_server → agent_brain_server}/models/index.py +0 -0
- {doc_serve_server → agent_brain_server}/models/query.py +0 -0
- {doc_serve_server → agent_brain_server}/project_root.py +0 -0
- {doc_serve_server → agent_brain_server}/runtime.py +0 -0
- {doc_serve_server → agent_brain_server}/services/__init__.py +0 -0
- {doc_serve_server → agent_brain_server}/storage/__init__.py +0 -0
- {doc_serve_server → agent_brain_server}/storage_paths.py +0 -0
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
Metadata-Version: 2.3
|
|
2
2
|
Name: agent-brain-rag
|
|
3
|
-
Version: 1.
|
|
4
|
-
Summary: RAG-
|
|
3
|
+
Version: 1.2.0
|
|
4
|
+
Summary: Agent Brain RAG - Intelligent document indexing and semantic search server that gives AI agents long-term memory
|
|
5
5
|
License: MIT
|
|
6
|
-
Keywords: rag,semantic-search,documentation,indexing,llama-index,chromadb,ai-agent,
|
|
6
|
+
Keywords: agent-brain,rag,semantic-search,ai-memory,llm-memory,documentation,indexing,llama-index,chromadb,ai-agent,claude-code,agent-memory
|
|
7
7
|
Author: Spillwave Solutions
|
|
8
8
|
Requires-Python: >=3.10,<4.0
|
|
9
9
|
Classifier: Development Status :: 4 - Beta
|
|
@@ -38,14 +38,14 @@ Project-URL: Homepage, https://github.com/SpillwaveSolutions/doc-serve
|
|
|
38
38
|
Project-URL: Repository, https://github.com/SpillwaveSolutions/doc-serve
|
|
39
39
|
Description-Content-Type: text/markdown
|
|
40
40
|
|
|
41
|
-
#
|
|
41
|
+
# Agent Brain RAG Server
|
|
42
42
|
|
|
43
|
-
|
|
43
|
+
Intelligent document indexing and semantic search REST API service that gives AI agents long-term memory.
|
|
44
44
|
|
|
45
45
|
## Installation
|
|
46
46
|
|
|
47
47
|
```bash
|
|
48
|
-
pip install
|
|
48
|
+
pip install agent-brain-rag
|
|
49
49
|
```
|
|
50
50
|
|
|
51
51
|
## Quick Start
|
|
@@ -58,15 +58,18 @@ pip install doc-serve
|
|
|
58
58
|
|
|
59
59
|
2. Start the server:
|
|
60
60
|
```bash
|
|
61
|
-
|
|
61
|
+
agent-brain-serve
|
|
62
62
|
```
|
|
63
63
|
|
|
64
64
|
The server will start at `http://127.0.0.1:8000`.
|
|
65
65
|
|
|
66
|
+
> **Note**: The legacy command `doc-serve` is still available but deprecated. Please use `agent-brain-serve` for new installations.
|
|
67
|
+
|
|
66
68
|
## Features
|
|
67
69
|
|
|
68
70
|
- **Document Indexing**: Load and index documents from folders (PDF, Markdown, TXT, DOCX, HTML)
|
|
69
|
-
- **
|
|
71
|
+
- **AST-Aware Code Ingestion**: Smart parsing for Python, TypeScript, JavaScript, Java, Go, Rust, C, C++
|
|
72
|
+
- **Hybrid Search**: Combines BM25 keyword matching with semantic similarity
|
|
70
73
|
- **Semantic Search**: Query indexed documents using natural language
|
|
71
74
|
- **OpenAI Embeddings**: Uses `text-embedding-3-large` for high-quality embeddings
|
|
72
75
|
- **Chroma Vector Store**: Persistent, thread-safe vector database
|
|
@@ -103,10 +106,10 @@ Required environment variables:
|
|
|
103
106
|
|
|
104
107
|
```bash
|
|
105
108
|
# Development mode
|
|
106
|
-
poetry run uvicorn
|
|
109
|
+
poetry run uvicorn agent_brain_server.api.main:app --reload
|
|
107
110
|
|
|
108
111
|
# Or use the entry point
|
|
109
|
-
poetry run
|
|
112
|
+
poetry run agent-brain-serve
|
|
110
113
|
```
|
|
111
114
|
|
|
112
115
|
The server will start at `http://127.0.0.1:8000`.
|
|
@@ -157,7 +160,7 @@ curl -X POST http://localhost:8000/query \
|
|
|
157
160
|
## Architecture
|
|
158
161
|
|
|
159
162
|
```
|
|
160
|
-
|
|
163
|
+
agent_brain_server/
|
|
161
164
|
├── api/
|
|
162
165
|
│ ├── main.py # FastAPI application
|
|
163
166
|
│ └── routers/ # Endpoint handlers
|
|
@@ -186,14 +189,14 @@ poetry run pytest
|
|
|
186
189
|
### Code Formatting
|
|
187
190
|
|
|
188
191
|
```bash
|
|
189
|
-
poetry run black
|
|
190
|
-
poetry run ruff check
|
|
192
|
+
poetry run black agent_brain_server/
|
|
193
|
+
poetry run ruff check agent_brain_server/
|
|
191
194
|
```
|
|
192
195
|
|
|
193
196
|
### Type Checking
|
|
194
197
|
|
|
195
198
|
```bash
|
|
196
|
-
poetry run mypy
|
|
199
|
+
poetry run mypy agent_brain_server/
|
|
197
200
|
```
|
|
198
201
|
|
|
199
202
|
## License
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
agent_brain_server/__init__.py,sha256=Mv_H50b2L5Gz9er1hpu_YZB3CCazw-I7QJgg9ZUgjgM,95
|
|
2
|
+
agent_brain_server/api/__init__.py,sha256=nvTvO_ahHAAsRDlV3dL_JlNruSdan4kav_P5sT_1PFk,93
|
|
3
|
+
agent_brain_server/api/main.py,sha256=lz0AngQ57zpZhscxE2QuHwt_4d6yclofo62Tdv3CPhQ,11752
|
|
4
|
+
agent_brain_server/api/routers/__init__.py,sha256=Z3PUKDwxeI8T88xMQeTcQ8nq1bQnu0nYjRnqNIsDUyY,254
|
|
5
|
+
agent_brain_server/api/routers/health.py,sha256=KmXkyoJDVLYakEmcdfjdqklxhRbdrhMGhnMsfeNgwzM,3491
|
|
6
|
+
agent_brain_server/api/routers/index.py,sha256=jTfeY8EIbM1tksJY6jNDw6qkmKBMaAi1ufDIdrg80WQ,6756
|
|
7
|
+
agent_brain_server/api/routers/query.py,sha256=TAbpuCybjacUo_-7Zm3Jt8EW8EMDTYJDLv6aGtjaUzQ,2793
|
|
8
|
+
agent_brain_server/config/__init__.py,sha256=zzDErZGUBwUm5Fk43OHJN7eWpeIw_1kWdnhsN6QQqSc,84
|
|
9
|
+
agent_brain_server/config/settings.py,sha256=ePpGSJ2VVov_LOtQtaOedi_HM-Yf8YcpJQzSyLHfyZQ,2629
|
|
10
|
+
agent_brain_server/indexing/__init__.py,sha256=6ISRNpfpokbI1x9bo8_SIRs9xx3x3YlYgeWYWHQj0oI,595
|
|
11
|
+
agent_brain_server/indexing/bm25_index.py,sha256=CIeGZIa8PwvOm_Fnz3EMgETvM2Mu6GzANWeUims0P0Q,5268
|
|
12
|
+
agent_brain_server/indexing/chunking.py,sha256=kW-ZOdwOS-PfXuPAYb8qL4-7yAY-rpY9a1el2YCN9CA,29848
|
|
13
|
+
agent_brain_server/indexing/document_loader.py,sha256=D6U3mDl_2jFfT_JxD-5yudtqlJW8Fh0zj5lhz0V0tPs,16090
|
|
14
|
+
agent_brain_server/indexing/embedding.py,sha256=JqiE9AOSooMSsALYtx7F1KFvjYVWwJt2dW-TpNmgGBw,9055
|
|
15
|
+
agent_brain_server/locking.py,sha256=yRpswpBem4964gTh1VH4VUPT-vBIRnNEVgWdorJA4Hg,3365
|
|
16
|
+
agent_brain_server/models/__init__.py,sha256=JLVsiVet-JDVC9s6ZefBcTXWZT5TVXrbI0E2B3ejpWM,478
|
|
17
|
+
agent_brain_server/models/health.py,sha256=54S5_uS7Ca24heQcMmRZn-fTwNNVpMDuDLE1EQwqfHE,3661
|
|
18
|
+
agent_brain_server/models/index.py,sha256=pjDv7phLS6dpiHLlEtcAuXQN_aHIfA_4lMkAZ-NkXZQ,5400
|
|
19
|
+
agent_brain_server/models/query.py,sha256=phl-bJFxhu83k1iP4Cx3GH5qJUQ12Y1Ah_bAwXOJVZM,6054
|
|
20
|
+
agent_brain_server/project_root.py,sha256=HIY5NMRDYWYIT7K6B_UMOGo1zXn9zwAdGI6ApViKI_8,2194
|
|
21
|
+
agent_brain_server/runtime.py,sha256=bcchfDBt_tn2_r-lFhxqKp0RncKrz152D-oM5RcKGrU,3076
|
|
22
|
+
agent_brain_server/services/__init__.py,sha256=E4VPN9Rqa2mxGQQEQn-5IYj63LSPTrA8aIx8ENO5xcc,296
|
|
23
|
+
agent_brain_server/services/indexing_service.py,sha256=ichWj0z-PaIWLMmSp85mkE80J7JjpzmRx6-jT3aA-BU,18563
|
|
24
|
+
agent_brain_server/services/query_service.py,sha256=RwU6eBxK0AhA0w4ZQzNxYUz_NgMGHW5yTMmKY0KTtgw,14881
|
|
25
|
+
agent_brain_server/storage/__init__.py,sha256=T4N0DhDT3av7zn3Ra6uLhMsjL9N5wakqC4cl-QXWvmM,222
|
|
26
|
+
agent_brain_server/storage/vector_store.py,sha256=gVaOJ9B9eRWYkteBoinb6vQHsKuRQEm8-vTo-nXuDBU,10883
|
|
27
|
+
agent_brain_server/storage_paths.py,sha256=aHSsTwELwIk1VOLO0L6O-RsbRJ1U20jvJ-EQsFKG1a8,1761
|
|
28
|
+
agent_brain_rag-1.2.0.dist-info/METADATA,sha256=LilsuwE92JbI_FJh9cYXhqhe2IdRnAfm7ijdQKEwi0s,5626
|
|
29
|
+
agent_brain_rag-1.2.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
|
|
30
|
+
agent_brain_rag-1.2.0.dist-info/entry_points.txt,sha256=r6kfBauVV9OtMXQ7trM9ofbW5N_kdzqTouoJ9zc31-I,122
|
|
31
|
+
agent_brain_rag-1.2.0.dist-info/RECORD,,
|
|
@@ -1,8 +1,15 @@
|
|
|
1
|
-
"""FastAPI application entry point.
|
|
1
|
+
"""FastAPI application entry point.
|
|
2
|
+
|
|
3
|
+
This module provides the Agent Brain RAG server, a FastAPI application
|
|
4
|
+
for document indexing and semantic search. The primary entry point is
|
|
5
|
+
`agent-brain-serve`, with `doc-serve` provided for backward compatibility.
|
|
6
|
+
"""
|
|
2
7
|
|
|
3
8
|
import logging
|
|
4
9
|
import os
|
|
5
10
|
import socket
|
|
11
|
+
import sys
|
|
12
|
+
import warnings
|
|
6
13
|
from collections.abc import AsyncIterator
|
|
7
14
|
from contextlib import asynccontextmanager
|
|
8
15
|
from pathlib import Path
|
|
@@ -13,15 +20,20 @@ import uvicorn
|
|
|
13
20
|
from fastapi import FastAPI
|
|
14
21
|
from fastapi.middleware.cors import CORSMiddleware
|
|
15
22
|
|
|
16
|
-
from
|
|
17
|
-
from
|
|
18
|
-
from
|
|
19
|
-
from
|
|
20
|
-
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
|
|
23
|
+
from agent_brain_server import __version__
|
|
24
|
+
from agent_brain_server.config import settings
|
|
25
|
+
from agent_brain_server.indexing.bm25_index import BM25IndexManager
|
|
26
|
+
from agent_brain_server.locking import (
|
|
27
|
+
acquire_lock,
|
|
28
|
+
cleanup_stale,
|
|
29
|
+
is_stale,
|
|
30
|
+
release_lock,
|
|
31
|
+
)
|
|
32
|
+
from agent_brain_server.project_root import resolve_project_root
|
|
33
|
+
from agent_brain_server.runtime import RuntimeState, delete_runtime, write_runtime
|
|
34
|
+
from agent_brain_server.services import IndexingService, QueryService
|
|
35
|
+
from agent_brain_server.storage import VectorStoreManager
|
|
36
|
+
from agent_brain_server.storage_paths import resolve_state_dir, resolve_storage_paths
|
|
25
37
|
|
|
26
38
|
from .routers import health_router, index_router, query_router
|
|
27
39
|
|
|
@@ -52,7 +64,7 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
|
|
|
52
64
|
"""
|
|
53
65
|
global _runtime_state, _state_dir
|
|
54
66
|
|
|
55
|
-
logger.info("Starting
|
|
67
|
+
logger.info("Starting Agent Brain RAG server...")
|
|
56
68
|
|
|
57
69
|
if settings.OPENAI_API_KEY:
|
|
58
70
|
os.environ["OPENAI_API_KEY"] = settings.OPENAI_API_KEY
|
|
@@ -138,7 +150,7 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
|
|
|
138
150
|
|
|
139
151
|
yield
|
|
140
152
|
|
|
141
|
-
logger.info("Shutting down
|
|
153
|
+
logger.info("Shutting down Agent Brain RAG server...")
|
|
142
154
|
|
|
143
155
|
# Cleanup for per-project mode
|
|
144
156
|
if state_dir is not None:
|
|
@@ -149,12 +161,12 @@ async def lifespan(app: FastAPI) -> AsyncIterator[None]:
|
|
|
149
161
|
|
|
150
162
|
# Create FastAPI application
|
|
151
163
|
app = FastAPI(
|
|
152
|
-
title="
|
|
164
|
+
title="Agent Brain RAG API",
|
|
153
165
|
description=(
|
|
154
166
|
"RAG-based document indexing and semantic search API. "
|
|
155
167
|
"Index documents from folders and query them using natural language."
|
|
156
168
|
),
|
|
157
|
-
version="1.
|
|
169
|
+
version="1.2.0",
|
|
158
170
|
lifespan=lifespan,
|
|
159
171
|
docs_url="/docs",
|
|
160
172
|
redoc_url="/redoc",
|
|
@@ -180,8 +192,8 @@ app.include_router(query_router, prefix="/query", tags=["Querying"])
|
|
|
180
192
|
async def root() -> dict[str, str]:
|
|
181
193
|
"""Root endpoint redirects to docs."""
|
|
182
194
|
return {
|
|
183
|
-
"name": "
|
|
184
|
-
"version": "1.
|
|
195
|
+
"name": "Agent Brain RAG API",
|
|
196
|
+
"version": "1.2.0",
|
|
185
197
|
"docs": "/docs",
|
|
186
198
|
"health": "/health",
|
|
187
199
|
}
|
|
@@ -246,7 +258,7 @@ def run(
|
|
|
246
258
|
logger.info(f"Per-project mode enabled: {_state_dir}")
|
|
247
259
|
|
|
248
260
|
uvicorn.run(
|
|
249
|
-
"
|
|
261
|
+
"agent_brain_server.api.main:app",
|
|
250
262
|
host=resolved_host,
|
|
251
263
|
port=resolved_port,
|
|
252
264
|
reload=reload if reload is not None else settings.DEBUG,
|
|
@@ -254,7 +266,7 @@ def run(
|
|
|
254
266
|
|
|
255
267
|
|
|
256
268
|
@click.command()
|
|
257
|
-
@click.version_option(version=__version__, prog_name="
|
|
269
|
+
@click.version_option(version=__version__, prog_name="agent-brain-serve")
|
|
258
270
|
@click.option(
|
|
259
271
|
"--host",
|
|
260
272
|
"-h",
|
|
@@ -292,19 +304,19 @@ def cli(
|
|
|
292
304
|
state_dir: Optional[str],
|
|
293
305
|
project_dir: Optional[str],
|
|
294
306
|
) -> None:
|
|
295
|
-
"""
|
|
307
|
+
"""Agent Brain RAG Server - Document indexing and semantic search API.
|
|
296
308
|
|
|
297
309
|
Start the FastAPI server for document indexing and querying.
|
|
298
310
|
|
|
299
311
|
\b
|
|
300
312
|
Examples:
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
313
|
+
agent-brain-serve # Start with default settings
|
|
314
|
+
agent-brain-serve --port 8080 # Start on port 8080
|
|
315
|
+
agent-brain-serve --port 0 # Auto-assign an available port
|
|
316
|
+
agent-brain-serve --host 0.0.0.0 # Bind to all interfaces
|
|
317
|
+
agent-brain-serve --reload # Enable auto-reload
|
|
318
|
+
agent-brain-serve --project-dir /my/project # Per-project mode
|
|
319
|
+
agent-brain-serve --state-dir /path/.claude/doc-serve # Explicit state dir
|
|
308
320
|
|
|
309
321
|
\b
|
|
310
322
|
Environment Variables:
|
|
@@ -328,5 +340,29 @@ def cli(
|
|
|
328
340
|
run(host=host, port=port, reload=reload, state_dir=resolved_state_dir)
|
|
329
341
|
|
|
330
342
|
|
|
343
|
+
def cli_deprecated() -> None:
|
|
344
|
+
"""Deprecated entry point for doc-serve command.
|
|
345
|
+
|
|
346
|
+
Shows a deprecation warning and then runs the main CLI.
|
|
347
|
+
"""
|
|
348
|
+
warnings.warn(
|
|
349
|
+
"\n"
|
|
350
|
+
"WARNING: 'doc-serve' is deprecated and will be removed in v2.0.\n"
|
|
351
|
+
"Please use 'agent-brain-serve' instead.\n"
|
|
352
|
+
"\n"
|
|
353
|
+
"Migration guide: docs/MIGRATION.md\n"
|
|
354
|
+
"Online: https://github.com/SpillwaveSolutions/agent-brain/blob/main/docs/MIGRATION.md\n",
|
|
355
|
+
DeprecationWarning,
|
|
356
|
+
stacklevel=1,
|
|
357
|
+
)
|
|
358
|
+
# Print to stderr for visibility since warnings may be filtered
|
|
359
|
+
print(
|
|
360
|
+
"\033[93mWARNING: 'doc-serve' is deprecated. "
|
|
361
|
+
"Use 'agent-brain-serve' instead. See docs/MIGRATION.md\033[0m",
|
|
362
|
+
file=sys.stderr,
|
|
363
|
+
)
|
|
364
|
+
cli()
|
|
365
|
+
|
|
366
|
+
|
|
331
367
|
if __name__ == "__main__":
|
|
332
368
|
cli()
|
|
@@ -5,7 +5,8 @@ from typing import Literal
|
|
|
5
5
|
|
|
6
6
|
from fastapi import APIRouter, Request
|
|
7
7
|
|
|
8
|
-
from
|
|
8
|
+
from agent_brain_server import __version__
|
|
9
|
+
from agent_brain_server.models import HealthStatus, IndexingStatus
|
|
9
10
|
|
|
10
11
|
router = APIRouter()
|
|
11
12
|
|
|
@@ -54,7 +55,7 @@ async def health_check(request: Request) -> HealthStatus:
|
|
|
54
55
|
status=status,
|
|
55
56
|
message=message,
|
|
56
57
|
timestamp=datetime.now(timezone.utc),
|
|
57
|
-
version=
|
|
58
|
+
version=__version__,
|
|
58
59
|
mode=mode,
|
|
59
60
|
instance_id=instance_id,
|
|
60
61
|
project_id=project_id,
|
|
@@ -4,7 +4,7 @@ import logging
|
|
|
4
4
|
|
|
5
5
|
from fastapi import APIRouter, HTTPException, Request, status
|
|
6
6
|
|
|
7
|
-
from
|
|
7
|
+
from agent_brain_server.models import QueryRequest, QueryResponse
|
|
8
8
|
|
|
9
9
|
logger = logging.getLogger(__name__)
|
|
10
10
|
|
|
@@ -33,8 +33,8 @@ async def query_documents(
|
|
|
33
33
|
400: Invalid query (empty or too long)
|
|
34
34
|
503: Index not ready (indexing in progress or not initialized)
|
|
35
35
|
"""
|
|
36
|
-
from
|
|
37
|
-
from
|
|
36
|
+
from agent_brain_server.services import QueryService
|
|
37
|
+
from agent_brain_server.services.indexing_service import IndexingService
|
|
38
38
|
|
|
39
39
|
query_service: QueryService = request.app.state.query_service
|
|
40
40
|
indexing_service: IndexingService = request.app.state.indexing_service
|
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
"""Indexing pipeline components for document processing."""
|
|
2
2
|
|
|
3
|
-
from
|
|
4
|
-
from
|
|
5
|
-
from
|
|
6
|
-
from
|
|
3
|
+
from agent_brain_server.indexing.bm25_index import BM25IndexManager, get_bm25_manager
|
|
4
|
+
from agent_brain_server.indexing.chunking import CodeChunker, ContextAwareChunker
|
|
5
|
+
from agent_brain_server.indexing.document_loader import DocumentLoader
|
|
6
|
+
from agent_brain_server.indexing.embedding import (
|
|
7
7
|
EmbeddingGenerator,
|
|
8
8
|
get_embedding_generator,
|
|
9
9
|
)
|
|
@@ -8,7 +8,7 @@ from typing import Optional
|
|
|
8
8
|
from llama_index.core.schema import BaseNode, NodeWithScore
|
|
9
9
|
from llama_index.retrievers.bm25 import BM25Retriever
|
|
10
10
|
|
|
11
|
-
from
|
|
11
|
+
from agent_brain_server.config import settings
|
|
12
12
|
|
|
13
13
|
logger = logging.getLogger(__name__)
|
|
14
14
|
|
|
@@ -13,7 +13,7 @@ import tree_sitter
|
|
|
13
13
|
import tree_sitter_language_pack as tslp
|
|
14
14
|
from llama_index.core.node_parser import CodeSplitter, SentenceSplitter
|
|
15
15
|
|
|
16
|
-
from
|
|
16
|
+
from agent_brain_server.config import settings
|
|
17
17
|
|
|
18
18
|
from .document_loader import LoadedDocument
|
|
19
19
|
|
|
@@ -22,7 +22,7 @@ class HealthStatus(BaseModel):
|
|
|
22
22
|
description="Timestamp of the health check",
|
|
23
23
|
)
|
|
24
24
|
version: str = Field(
|
|
25
|
-
default="1.
|
|
25
|
+
default="1.2.0",
|
|
26
26
|
description="Server version",
|
|
27
27
|
)
|
|
28
28
|
mode: Optional[str] = Field(
|
|
@@ -49,7 +49,7 @@ class HealthStatus(BaseModel):
|
|
|
49
49
|
"status": "healthy",
|
|
50
50
|
"message": "Server is running and ready for queries",
|
|
51
51
|
"timestamp": "2024-12-15T10:30:00Z",
|
|
52
|
-
"version": "1.
|
|
52
|
+
"version": "1.2.0",
|
|
53
53
|
}
|
|
54
54
|
]
|
|
55
55
|
}
|
|
@@ -10,16 +10,16 @@ from typing import Any, Callable, Optional, Union
|
|
|
10
10
|
|
|
11
11
|
from llama_index.core.schema import TextNode
|
|
12
12
|
|
|
13
|
-
from
|
|
13
|
+
from agent_brain_server.indexing import (
|
|
14
14
|
BM25IndexManager,
|
|
15
15
|
ContextAwareChunker,
|
|
16
16
|
DocumentLoader,
|
|
17
17
|
EmbeddingGenerator,
|
|
18
18
|
get_bm25_manager,
|
|
19
19
|
)
|
|
20
|
-
from
|
|
21
|
-
from
|
|
22
|
-
from
|
|
20
|
+
from agent_brain_server.indexing.chunking import CodeChunk, CodeChunker, TextChunk
|
|
21
|
+
from agent_brain_server.models import IndexingState, IndexingStatusEnum, IndexRequest
|
|
22
|
+
from agent_brain_server.storage import VectorStoreManager, get_vector_store
|
|
23
23
|
|
|
24
24
|
logger = logging.getLogger(__name__)
|
|
25
25
|
|
|
@@ -7,10 +7,15 @@ from typing import Any, Optional
|
|
|
7
7
|
from llama_index.core.retrievers import BaseRetriever
|
|
8
8
|
from llama_index.core.schema import NodeWithScore, QueryBundle, TextNode
|
|
9
9
|
|
|
10
|
-
from
|
|
11
|
-
from
|
|
12
|
-
from
|
|
13
|
-
|
|
10
|
+
from agent_brain_server.indexing import EmbeddingGenerator, get_embedding_generator
|
|
11
|
+
from agent_brain_server.indexing.bm25_index import BM25IndexManager, get_bm25_manager
|
|
12
|
+
from agent_brain_server.models import (
|
|
13
|
+
QueryMode,
|
|
14
|
+
QueryRequest,
|
|
15
|
+
QueryResponse,
|
|
16
|
+
QueryResult,
|
|
17
|
+
)
|
|
18
|
+
from agent_brain_server.storage import VectorStoreManager, get_vector_store
|
|
14
19
|
|
|
15
20
|
logger = logging.getLogger(__name__)
|
|
16
21
|
|
|
@@ -1,31 +0,0 @@
|
|
|
1
|
-
doc_serve_server/__init__.py,sha256=CWV76BF1YvkQVzexuEm1sl4Zn4z27wRd_9dWbAZNWa0,95
|
|
2
|
-
doc_serve_server/api/__init__.py,sha256=nvTvO_ahHAAsRDlV3dL_JlNruSdan4kav_P5sT_1PFk,93
|
|
3
|
-
doc_serve_server/api/main.py,sha256=a3g_N9VEY-VAG-HvNXMW7u-xWvwjP3P9BtPlxNgC900,10628
|
|
4
|
-
doc_serve_server/api/routers/__init__.py,sha256=Z3PUKDwxeI8T88xMQeTcQ8nq1bQnu0nYjRnqNIsDUyY,254
|
|
5
|
-
doc_serve_server/api/routers/health.py,sha256=cbiBooeKMKoTpje8xd2Yywpr7fSrCJZP9ubl8EJOjEo,3442
|
|
6
|
-
doc_serve_server/api/routers/index.py,sha256=M4zOlXX_pmMYDcmfK6fh9PX5qqw-7xHuwG-Mp-0CZw0,6754
|
|
7
|
-
doc_serve_server/api/routers/query.py,sha256=zHyzQBPOz2Xz8MAF1rU2reWopYJsXCBWFW2kQe_f4hg,2787
|
|
8
|
-
doc_serve_server/config/__init__.py,sha256=zzDErZGUBwUm5Fk43OHJN7eWpeIw_1kWdnhsN6QQqSc,84
|
|
9
|
-
doc_serve_server/config/settings.py,sha256=ePpGSJ2VVov_LOtQtaOedi_HM-Yf8YcpJQzSyLHfyZQ,2629
|
|
10
|
-
doc_serve_server/indexing/__init__.py,sha256=7P1zcYAQFO4ME6uLtc75LL-GT7dEzXk2yY0JAcP0kRc,587
|
|
11
|
-
doc_serve_server/indexing/bm25_index.py,sha256=0JZ4_T1d9H7FLUDqnnXHcoO2_qrztzXrtNj7HNMg2WQ,5266
|
|
12
|
-
doc_serve_server/indexing/chunking.py,sha256=uGeCRRT19eVcFPpK9KkxI0_1OXjFRQZZI6B5CwCVR-4,29846
|
|
13
|
-
doc_serve_server/indexing/document_loader.py,sha256=D6U3mDl_2jFfT_JxD-5yudtqlJW8Fh0zj5lhz0V0tPs,16090
|
|
14
|
-
doc_serve_server/indexing/embedding.py,sha256=MtL9SLpBQxh99hJ_dufah0xyIGwznHGwnfnqcdyhSCc,9053
|
|
15
|
-
doc_serve_server/locking.py,sha256=yRpswpBem4964gTh1VH4VUPT-vBIRnNEVgWdorJA4Hg,3365
|
|
16
|
-
doc_serve_server/models/__init__.py,sha256=JLVsiVet-JDVC9s6ZefBcTXWZT5TVXrbI0E2B3ejpWM,478
|
|
17
|
-
doc_serve_server/models/health.py,sha256=-NWwyPI0cUK1Sqhf5x0WO-ilyJbwJxvROzeaFn12WDc,3661
|
|
18
|
-
doc_serve_server/models/index.py,sha256=pjDv7phLS6dpiHLlEtcAuXQN_aHIfA_4lMkAZ-NkXZQ,5400
|
|
19
|
-
doc_serve_server/models/query.py,sha256=phl-bJFxhu83k1iP4Cx3GH5qJUQ12Y1Ah_bAwXOJVZM,6054
|
|
20
|
-
doc_serve_server/project_root.py,sha256=HIY5NMRDYWYIT7K6B_UMOGo1zXn9zwAdGI6ApViKI_8,2194
|
|
21
|
-
doc_serve_server/runtime.py,sha256=bcchfDBt_tn2_r-lFhxqKp0RncKrz152D-oM5RcKGrU,3076
|
|
22
|
-
doc_serve_server/services/__init__.py,sha256=E4VPN9Rqa2mxGQQEQn-5IYj63LSPTrA8aIx8ENO5xcc,296
|
|
23
|
-
doc_serve_server/services/indexing_service.py,sha256=rOs0xlMy_1RQuoPt62w428-CshPWj91CFSpkL251TEI,18555
|
|
24
|
-
doc_serve_server/services/query_service.py,sha256=elyKWl02v8BzjdBr8S7KOiUUEdS2Rp3V85ifUU5Bi-g,14852
|
|
25
|
-
doc_serve_server/storage/__init__.py,sha256=T4N0DhDT3av7zn3Ra6uLhMsjL9N5wakqC4cl-QXWvmM,222
|
|
26
|
-
doc_serve_server/storage/vector_store.py,sha256=9pdEPvlf2JAUnwxDpb9CwY_A7sIqv2iu5ddy5SQm-ys,10881
|
|
27
|
-
doc_serve_server/storage_paths.py,sha256=aHSsTwELwIk1VOLO0L6O-RsbRJ1U20jvJ-EQsFKG1a8,1761
|
|
28
|
-
agent_brain_rag-1.1.0.dist-info/METADATA,sha256=dOzH_3y00uehax1LiI5QBHToTwnN09tGyoRygBw_uVA,5230
|
|
29
|
-
agent_brain_rag-1.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
|
|
30
|
-
agent_brain_rag-1.1.0.dist-info/entry_points.txt,sha256=DOPhlYyScH9BTPXr9_FJzgRplzHe6bR9q2pTCCg2QOI,59
|
|
31
|
-
agent_brain_rag-1.1.0.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|