Fix RAG inference

This commit is contained in:
2025-08-07 15:38:01 -05:00
parent 4327e1c581
commit e504407190
9 changed files with 222 additions and 31 deletions
+2 -2
View File
@@ -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.