File size: 478 Bytes
db2df31
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
"""
Centralized Chroma configuration.

The project intentionally uses a single persistent Chroma collection and separates
different logical stores (KB docs, Tavily web cache, semantic cache, etc.) via a
`namespace` metadata field.
"""

from __future__ import annotations

import os

PERSIST_DIRECTORY = "src/data/.chroma"
COLLECTION_NAME = "finance_assistant"


def ensure_persist_dir(path: str = PERSIST_DIRECTORY) -> str:
    os.makedirs(path, exist_ok=True)
    return path