diff --git a/src/wandbot/evaluation/eval/async_main.py b/src/wandbot/evaluation/eval/async_main.py index 13fa49a..d219d5c 100644 --- a/src/wandbot/evaluation/eval/async_main.py +++ b/src/wandbot/evaluation/eval/async_main.py @@ -7,10 +7,11 @@ import aiofiles import httpx import pandas as pd -import wandb from llama_index.llms.openai import OpenAI from tenacity import retry, stop_after_attempt, wait_random_exponential from tqdm import tqdm + +import wandb from wandbot.evaluation.config import EvalConfig from wandbot.evaluation.eval.correctness import ( CORRECTNESS_EVAL_TEMPLATE, diff --git a/src/wandbot/rag/retrieval.py b/src/wandbot/rag/retrieval.py index 065c791..1a72ac9 100644 --- a/src/wandbot/rag/retrieval.py +++ b/src/wandbot/rag/retrieval.py @@ -3,6 +3,7 @@ from langchain.retrievers.document_compressors import CohereRerank from langchain_core.documents import Document from langchain_core.runnables import Runnable, RunnablePassthrough + from wandbot.rag.utils import get_web_contexts from wandbot.retriever.base import VectorStore from wandbot.retriever.web_search import YouSearch, YouSearchConfig