Hugging Face's logo

Spaces:
FD900
/
Final_Assignment_Template
Sleeping

App Files Community
1
FD900 commited on
Commit
db208f0
·
verified ·
1 Parent(s): 1ddbbf3

Update tools/content_retriever_tool.py

Browse files
Files changed (1) hide show
  1. tools/content_retriever_tool.py +1 -1
tools/content_retriever_tool.py CHANGED
@@ -4,7 +4,7 @@ from docling.chunking import HierarchicalChunker
4
  from sentence_transformers import SentenceTransformer, util
5
  import torch
6
 
7
- class ContentRetrieverTool(Tool):
8
  name = 'retrieve_content'
9
  description = "Extracts relevant content from a file or URL (PDF, DOCX, XLSX, HTML, etc.) based on a given query."
10
 
 
4
  from sentence_transformers import SentenceTransformer, util
5
  import torch
6
 
7
+ class ContentRetrieverTool(BaseTool):
8
  name = 'retrieve_content'
9
  description = "Extracts relevant content from a file or URL (PDF, DOCX, XLSX, HTML, etc.) based on a given query."
10