Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions bin/embeddings_manager
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import os
import re
import sys
from argparse import ArgumentParser
from pathlib import Path, PurePosixPath
from shutil import rmtree
Expand All @@ -12,6 +13,13 @@ import boto3
from botocore import UNSIGNED
from botocore.client import Config


# Run from anywhere without setting PYTHONPATH, like the other bin/ scripts.
# The Dockerfile and CI both set it (`/app/src`, `./bin:./src`), so importing
# worked there and nowhere else -- including in this script's own documented
# usage.
sys.path.insert(0, str(Path(__file__).resolve().parent.parent / "src"))

from data_generation.alliance import generate_alliance_embeddings
from data_generation.reactome import generate_reactome_embeddings
from data_generation.uniprot import generate_uniprot_embeddings
Expand Down
7 changes: 7 additions & 0 deletions bin/retrieval_baseline
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,13 @@ from langchain_core.documents import Document
from langchain_core.retrievers import BaseRetriever
from nltk.tokenize import word_tokenize


# Run from anywhere without setting PYTHONPATH, like the other bin/ scripts.
# The Dockerfile and CI both set it (`/app/src`, `./bin:./src`), so importing
# worked there and nowhere else -- including in this script's own documented
# usage.
sys.path.insert(0, str(Path(__file__).resolve().parent.parent / "src"))

from agent.models import get_embedding, get_llm
from retrievers.csv_chroma import (
VECTOR_OVERFETCH,
Expand Down
Loading