Ilia Tambovtsev commited on
Commit
b7dc0e7
·
1 Parent(s): 4c785d2

fix: implement resetting prompt

Browse files
Files changed (1) hide show
  1. src/chains/pipelines.py +5 -1
src/chains/pipelines.py CHANGED
@@ -29,6 +29,10 @@ class SlideAnalysis(BaseModel):
29
  vision_prompt: Optional[str]
30
  content: str
31
 
 
 
 
 
32
 
33
  class PresentationAnalysis(BaseModel):
34
  """Container for presentation analysis results"""
@@ -230,7 +234,7 @@ class PresentationPipeline(Chain):
230
  "page_num": page_num
231
  })
232
  slide_analysis = result["slide_analysis"]
233
- slide_analysis["vision_prompt"] = None
234
  return slide_analysis
235
  except Exception as e:
236
  logger.error(f"Failed to process slide {page_num}: {str(e)}")
 
29
  vision_prompt: Optional[str]
30
  content: str
31
 
32
+ def reset_vision_prompt(self):
33
+ """Reset vision prompt"""
34
+ self.vision_prompt = None
35
+
36
 
37
  class PresentationAnalysis(BaseModel):
38
  """Container for presentation analysis results"""
 
234
  "page_num": page_num
235
  })
236
  slide_analysis = result["slide_analysis"]
237
+ slide_analysis.reset_vision_prompt()
238
  return slide_analysis
239
  except Exception as e:
240
  logger.error(f"Failed to process slide {page_num}: {str(e)}")