hirumunasinghe commited on
Commit
71793d1
·
verified ·
1 Parent(s): a91323c

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. .gitignore +0 -1
  2. app/streamlit_app.py +0 -3
  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 IncludeEnum, Metadata
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]