Spaces:
Sleeping
Sleeping
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]:
|