Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- .gitignore +0 -1
- app/streamlit_app.py +0 -3
- src/vector_store.py +2 -6
.gitignore
CHANGED
|
@@ -18,7 +18,6 @@ outputs/
|
|
| 18 |
# Common binaries
|
| 19 |
*.sqlite3
|
| 20 |
*.bin
|
| 21 |
-
*.pdf
|
| 22 |
*.parquet
|
| 23 |
*.pt
|
| 24 |
*.onnx
|
|
|
|
| 18 |
# Common binaries
|
| 19 |
*.sqlite3
|
| 20 |
*.bin
|
|
|
|
| 21 |
*.parquet
|
| 22 |
*.pt
|
| 23 |
*.onnx
|
app/streamlit_app.py
CHANGED
|
@@ -19,10 +19,7 @@ if str(ROOT_DIR) not in sys.path:
|
|
| 19 |
sys.path.append(str(ROOT_DIR))
|
| 20 |
|
| 21 |
from src.models import StrategicObjective, ActionTask, load_actions, load_strategies
|
| 22 |
-
from src.alignment import AlignmentEngine
|
| 23 |
from src.recommendations import generate_recommendations
|
| 24 |
-
from src.ontology import build_graph_from_alignment, save_graph, query_graph_stats
|
| 25 |
-
from src.evaluation import run_evaluation
|
| 26 |
from src.rag_engine import RAGEngine
|
| 27 |
from src.pipeline import run_full_flow
|
| 28 |
from src.viz import (
|
|
|
|
| 19 |
sys.path.append(str(ROOT_DIR))
|
| 20 |
|
| 21 |
from src.models import StrategicObjective, ActionTask, load_actions, load_strategies
|
|
|
|
| 22 |
from src.recommendations import generate_recommendations
|
|
|
|
|
|
|
| 23 |
from src.rag_engine import RAGEngine
|
| 24 |
from src.pipeline import run_full_flow
|
| 25 |
from src.viz import (
|
src/vector_store.py
CHANGED
|
@@ -6,7 +6,7 @@ import logging
|
|
| 6 |
|
| 7 |
import chromadb
|
| 8 |
from chromadb.config import Settings
|
| 9 |
-
from chromadb.api.types import
|
| 10 |
import numpy as np
|
| 11 |
|
| 12 |
|
|
@@ -131,11 +131,7 @@ class ActionVectorStore:
|
|
| 131 |
res = self.collection.query(
|
| 132 |
query_embeddings=[list(embedding)],
|
| 133 |
n_results=top_k,
|
| 134 |
-
include=[
|
| 135 |
-
IncludeEnum.distances,
|
| 136 |
-
IncludeEnum.metadatas,
|
| 137 |
-
IncludeEnum.documents,
|
| 138 |
-
],
|
| 139 |
)
|
| 140 |
ids = (res.get("ids") or [[]])[0]
|
| 141 |
dists = (res.get("distances") or [[]])[0]
|
|
|
|
| 6 |
|
| 7 |
import chromadb
|
| 8 |
from chromadb.config import Settings
|
| 9 |
+
from chromadb.api.types import Metadata
|
| 10 |
import numpy as np
|
| 11 |
|
| 12 |
|
|
|
|
| 131 |
res = self.collection.query(
|
| 132 |
query_embeddings=[list(embedding)],
|
| 133 |
n_results=top_k,
|
| 134 |
+
include=["distances", "metadatas", "documents"],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 135 |
)
|
| 136 |
ids = (res.get("ids") or [[]])[0]
|
| 137 |
dists = (res.get("distances") or [[]])[0]
|