Spaces:
Sleeping
Sleeping
update hybrid_retriever_tool file
Browse files
tools/hybrid_retriever_tool.py
CHANGED
|
@@ -3,10 +3,10 @@ from rank_bm25 import BM25Okapi
|
|
| 3 |
from sentence_transformers import SentenceTransformer
|
| 4 |
from tavily import TavilyClient
|
| 5 |
from openai import OpenAI
|
| 6 |
-
from crewai_tools import
|
| 7 |
import os
|
| 8 |
|
| 9 |
-
class HybridRetrieverTool(
|
| 10 |
name = "Hybrid Retriever Tool"
|
| 11 |
description = "Combines BM25 keyword scoring with semantic similarity for hybrid retrieval"
|
| 12 |
|
|
|
|
| 3 |
from sentence_transformers import SentenceTransformer
|
| 4 |
from tavily import TavilyClient
|
| 5 |
from openai import OpenAI
|
| 6 |
+
from crewai_tools import RagTool
|
| 7 |
import os
|
| 8 |
|
| 9 |
+
class HybridRetrieverTool(RagTool):
|
| 10 |
name = "Hybrid Retriever Tool"
|
| 11 |
description = "Combines BM25 keyword scoring with semantic similarity for hybrid retrieval"
|
| 12 |
|