Fix RAG inference
This commit is contained in:
+2
-2
@@ -16,10 +16,10 @@ from tools.github_tool import GitHubTool
|
||||
# If you have downloaded a model, provide the local path here.
|
||||
# Otherwise, the model will be downloaded from Hugging Face.
|
||||
# Example: EMBEDDING_MODEL_PATH = "/path/to/your/models/all-MiniLM-L6-v2"
|
||||
EMBEDDING_MODEL_PATH = """C:\Models\embeddings\Qwen3-Embedding-0.6B"""
|
||||
EMBEDDING_MODEL_PATH = os.environ.get("EMBEDDING_MODEL_PATH")
|
||||
|
||||
# Path to store the local vector database
|
||||
CHROMA_DB_PATH = """C:\Models\embeddings\embedding_result\chroma_db"""
|
||||
CHROMA_DB_PATH = os.environ.get("CHROMA_DB_PATH")
|
||||
# Name of the collection within the database
|
||||
CHROMA_COLLECTION_NAME = "github_repo"
|
||||
# Files with these extensions will be indexed. Add any other text-based files you need.
|
||||
|
||||
Reference in New Issue
Block a user