Skip to content
Open
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
14 changes: 7 additions & 7 deletions learn2rag/evaluation/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@
import pandas as pd

import learn2rag.pipeline.ingestion
from learn2rag.pipeline.config import opt_config


def key_document(key: str) -> Callable[[Any], Any]:
Expand Down Expand Up @@ -117,9 +116,10 @@ def basic_pipeline(dataset_name: str, question: str) -> dict[str, Any]:
'imported_documents_file_path': None,
'llm': None,
}
documents = learn2rag.pipeline.search.search(question, user_config, opt_config)
answer = learn2rag.pipeline.generate.generate(question, documents.points, opt_config)
return {
'documents': documents,
'answer': answer,
}
raise NotImplementedError()
# documents = learn2rag.pipeline.search.search(question, user_config, opt_config)
# answer = learn2rag.pipeline.generate.generate(question, documents.points, opt_config)
# return {
# 'documents': documents,
# 'answer': answer,
# }
4 changes: 2 additions & 2 deletions learn2rag/pipeline/qdrant.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
import os

from typing import Any, Optional

import logging
from qdrant_client import QdrantClient
from qdrant_client.models import Distance, VectorParams, SparseVectorParams, SparseIndexParams, MultiVectorComparator, MultiVectorConfig
from qdrant_client.http.exceptions import UnexpectedResponse

from .config import user_config

# FIXME: when running a Windows package,
# this import causes segmentation fault if done after creating QdrantClient.
# Importing it here prevents this situation...
Expand Down
2 changes: 2 additions & 0 deletions learn2rag/ui/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@
normalize_path,
open_web_browser,
save_data_path,
get_default_rag_dir,
)

from datetime import datetime
Expand Down Expand Up @@ -191,6 +192,7 @@ def inject_info() -> dict[str, Any]:
'debug_logging': config.get('logging', {}).get('debug', False),
'current_timestamp': math.floor(time.time()),
'llm': learn2rag.pipeline.llm,
'default_rag_dir': get_default_rag_dir(),
}

@app.context_processor
Expand Down
Loading
Loading