Spaces:
Runtime error
Runtime error
Charles Azam
commited on
Commit
·
abd6d4e
1
Parent(s):
48a564a
feat: summary agent working
Browse files
src/deepengineer/deepsearch/analyse_markdown_agent.py
CHANGED
|
@@ -17,6 +17,7 @@ class GetTableOfContentsTool(Tool):
|
|
| 17 |
output_type = "string"
|
| 18 |
|
| 19 |
def __init__(self, markdown: OCRResponse):
|
|
|
|
| 20 |
self.markdown: OCRResponse = markdown
|
| 21 |
self.table_of_contents: str = get_table_of_contents_per_page_pdf(self.markdown)
|
| 22 |
|
|
@@ -30,6 +31,7 @@ class GetMarkdownTool(Tool):
|
|
| 30 |
output_type = "string"
|
| 31 |
|
| 32 |
def __init__(self, markdown: OCRResponse):
|
|
|
|
| 33 |
self.markdown: OCRResponse = markdown
|
| 34 |
self.markdown_content: str = convert_ocr_response_to_markdown(self.markdown)
|
| 35 |
|
|
@@ -49,6 +51,7 @@ class GetPagesContentTool(Tool):
|
|
| 49 |
output_type = "string"
|
| 50 |
|
| 51 |
def __init__(self, markdown: OCRResponse):
|
|
|
|
| 52 |
self.markdown: OCRResponse = markdown
|
| 53 |
|
| 54 |
def forward(self, page_numbers: list[int]) -> str:
|
|
@@ -66,6 +69,7 @@ class FindInPdfTool(Tool):
|
|
| 66 |
output_type = "array"
|
| 67 |
|
| 68 |
def __init__(self, markdown: OCRResponse):
|
|
|
|
| 69 |
self.markdown: OCRResponse = markdown
|
| 70 |
|
| 71 |
def forward(self, search_queries: list[str]) -> list[int]:
|
|
|
|
| 17 |
output_type = "string"
|
| 18 |
|
| 19 |
def __init__(self, markdown: OCRResponse):
|
| 20 |
+
super().__init__()
|
| 21 |
self.markdown: OCRResponse = markdown
|
| 22 |
self.table_of_contents: str = get_table_of_contents_per_page_pdf(self.markdown)
|
| 23 |
|
|
|
|
| 31 |
output_type = "string"
|
| 32 |
|
| 33 |
def __init__(self, markdown: OCRResponse):
|
| 34 |
+
super().__init__()
|
| 35 |
self.markdown: OCRResponse = markdown
|
| 36 |
self.markdown_content: str = convert_ocr_response_to_markdown(self.markdown)
|
| 37 |
|
|
|
|
| 51 |
output_type = "string"
|
| 52 |
|
| 53 |
def __init__(self, markdown: OCRResponse):
|
| 54 |
+
super().__init__()
|
| 55 |
self.markdown: OCRResponse = markdown
|
| 56 |
|
| 57 |
def forward(self, page_numbers: list[int]) -> str:
|
|
|
|
| 69 |
output_type = "array"
|
| 70 |
|
| 71 |
def __init__(self, markdown: OCRResponse):
|
| 72 |
+
super().__init__()
|
| 73 |
self.markdown: OCRResponse = markdown
|
| 74 |
|
| 75 |
def forward(self, search_queries: list[str]) -> list[int]:
|