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.
Files changed (33) hide show
  1. {agent_brain_rag-1.1.0.dist-info → agent_brain_rag-1.2.0.dist-info}/METADATA +17 -14
  2. agent_brain_rag-1.2.0.dist-info/RECORD +31 -0
  3. agent_brain_rag-1.2.0.dist-info/entry_points.txt +4 -0
  4. {doc_serve_server → agent_brain_server}/__init__.py +1 -1
  5. {doc_serve_server → agent_brain_server}/api/main.py +62 -26
  6. {doc_serve_server → agent_brain_server}/api/routers/health.py +3 -2
  7. {doc_serve_server → agent_brain_server}/api/routers/index.py +1 -1
  8. {doc_serve_server → agent_brain_server}/api/routers/query.py +3 -3
  9. {doc_serve_server → agent_brain_server}/indexing/__init__.py +4 -4
  10. {doc_serve_server → agent_brain_server}/indexing/bm25_index.py +1 -1
  11. {doc_serve_server → agent_brain_server}/indexing/chunking.py +1 -1
  12. {doc_serve_server → agent_brain_server}/indexing/embedding.py +1 -1
  13. {doc_serve_server → agent_brain_server}/models/health.py +2 -2
  14. {doc_serve_server → agent_brain_server}/services/indexing_service.py +4 -4
  15. {doc_serve_server → agent_brain_server}/services/query_service.py +9 -4
  16. {doc_serve_server → agent_brain_server}/storage/vector_store.py +1 -1
  17. agent_brain_rag-1.1.0.dist-info/RECORD +0 -31
  18. agent_brain_rag-1.1.0.dist-info/entry_points.txt +0 -3
  19. {agent_brain_rag-1.1.0.dist-info → agent_brain_rag-1.2.0.dist-info}/WHEEL +0 -0
  20. {doc_serve_server → agent_brain_server}/api/__init__.py +0 -0
  21. {doc_serve_server → agent_brain_server}/api/routers/__init__.py +0 -0
  22. {doc_serve_server → agent_brain_server}/config/__init__.py +0 -0
  23. {doc_serve_server → agent_brain_server}/config/settings.py +0 -0
  24. {doc_serve_server → agent_brain_server}/indexing/document_loader.py +0 -0
  25. {doc_serve_server → agent_brain_server}/locking.py +0 -0
  26. {doc_serve_server → agent_brain_server}/models/__init__.py +0 -0
  27. {doc_serve_server → agent_brain_server}/models/index.py +0 -0
  28. {doc_serve_server → agent_brain_server}/models/query.py +0 -0
  29. {doc_serve_server → agent_brain_server}/project_root.py +0 -0
  30. {doc_serve_server → agent_brain_server}/runtime.py +0 -0
  31. {doc_serve_server → agent_brain_server}/services/__init__.py +0 -0
  32. {doc_serve_server → agent_brain_server}/storage/__init__.py +0 -0
  33. {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.1.0
4
- Summary: RAG-based document indexing and semantic search server for AI agents
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,brain
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
- # Doc-Serve Server
41
+ # Agent Brain RAG Server
42
42
 
43
- RAG-based document indexing and semantic search REST API service.
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 doc-serve
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
- doc-serve
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
- - **Context-Aware Chunking**: Smart text splitting with configurable chunk sizes and overlap
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 doc_serve_server.api.main:app --reload
109
+ poetry run uvicorn agent_brain_server.api.main:app --reload
107
110
 
108
111
  # Or use the entry point
109
- poetry run doc-serve
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
- doc_serve_server/
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 doc_serve_server/
190
- poetry run ruff check doc_serve_server/
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 doc_serve_server/
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,,
@@ -0,0 +1,4 @@
1
+ [console_scripts]
2
+ agent-brain-serve=agent_brain_server.api.main:cli
3
+ doc-serve=agent_brain_server.api.main:cli_deprecated
4
+
@@ -1,3 +1,3 @@
1
1
  """Doc-Serve Server - RAG-based document indexing and query service."""
2
2
 
3
- __version__ = "1.1.0"
3
+ __version__ = "1.2.0"
@@ -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 doc_serve_server import __version__
17
- from doc_serve_server.config import settings
18
- from doc_serve_server.indexing.bm25_index import BM25IndexManager
19
- from doc_serve_server.locking import acquire_lock, cleanup_stale, is_stale, release_lock
20
- from doc_serve_server.project_root import resolve_project_root
21
- from doc_serve_server.runtime import RuntimeState, delete_runtime, write_runtime
22
- from doc_serve_server.services import IndexingService, QueryService
23
- from doc_serve_server.storage import VectorStoreManager
24
- from doc_serve_server.storage_paths import resolve_state_dir, resolve_storage_paths
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 Doc-Serve server...")
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 Doc-Serve server...")
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="Doc-Serve API",
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.1.0",
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": "Doc-Serve API",
184
- "version": "1.1.0",
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
- "doc_serve_server.api.main:app",
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="doc-serve")
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
- """Doc-Serve Server - RAG-based document indexing and semantic search API.
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
- doc-serve # Start with default settings
302
- doc-serve --port 8080 # Start on port 8080
303
- doc-serve --port 0 # Auto-assign an available port
304
- doc-serve --host 0.0.0.0 # Bind to all interfaces
305
- doc-serve --reload # Enable auto-reload
306
- doc-serve --project-dir /my/project # Per-project mode (auto state-dir)
307
- doc-serve --state-dir /path/.claude/doc-serve # Explicit state directory
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 doc_serve_server.models import HealthStatus, IndexingStatus
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="1.1.0",
58
+ version=__version__,
58
59
  mode=mode,
59
60
  instance_id=instance_id,
60
61
  project_id=project_id,
@@ -5,7 +5,7 @@ from pathlib import Path
5
5
 
6
6
  from fastapi import APIRouter, HTTPException, Request, status
7
7
 
8
- from doc_serve_server.models import IndexRequest, IndexResponse
8
+ from agent_brain_server.models import IndexRequest, IndexResponse
9
9
 
10
10
  router = APIRouter()
11
11
 
@@ -4,7 +4,7 @@ import logging
4
4
 
5
5
  from fastapi import APIRouter, HTTPException, Request, status
6
6
 
7
- from doc_serve_server.models import QueryRequest, QueryResponse
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 doc_serve_server.services import QueryService
37
- from doc_serve_server.services.indexing_service import IndexingService
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 doc_serve_server.indexing.bm25_index import BM25IndexManager, get_bm25_manager
4
- from doc_serve_server.indexing.chunking import CodeChunker, ContextAwareChunker
5
- from doc_serve_server.indexing.document_loader import DocumentLoader
6
- from doc_serve_server.indexing.embedding import (
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 doc_serve_server.config import settings
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 doc_serve_server.config import settings
16
+ from agent_brain_server.config import settings
17
17
 
18
18
  from .document_loader import LoadedDocument
19
19
 
@@ -7,7 +7,7 @@ from typing import Optional
7
7
  from anthropic import AsyncAnthropic
8
8
  from openai import AsyncOpenAI
9
9
 
10
- from doc_serve_server.config import settings
10
+ from agent_brain_server.config import settings
11
11
 
12
12
  from .chunking import TextChunk
13
13
 
@@ -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.1.0",
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.1.0",
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 doc_serve_server.indexing import (
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 doc_serve_server.indexing.chunking import CodeChunk, CodeChunker, TextChunk
21
- from doc_serve_server.models import IndexingState, IndexingStatusEnum, IndexRequest
22
- from doc_serve_server.storage import VectorStoreManager, get_vector_store
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 doc_serve_server.indexing import EmbeddingGenerator, get_embedding_generator
11
- from doc_serve_server.indexing.bm25_index import BM25IndexManager, get_bm25_manager
12
- from doc_serve_server.models import QueryMode, QueryRequest, QueryResponse, QueryResult
13
- from doc_serve_server.storage import VectorStoreManager, get_vector_store
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
 
@@ -9,7 +9,7 @@ from typing import Any, Optional
9
9
  import chromadb
10
10
  from chromadb.config import Settings as ChromaSettings
11
11
 
12
- from doc_serve_server.config import settings
12
+ from agent_brain_server.config import settings
13
13
 
14
14
  logger = logging.getLogger(__name__)
15
15
 
@@ -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,,
@@ -1,3 +0,0 @@
1
- [console_scripts]
2
- doc-serve=doc_serve_server.api.main:cli
3
-
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes