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