Ilia Tambovtsev commited on
Commit
a35a16c
·
1 Parent(s): 211933f

feat: save vision prompt to presentation

Browse files
Files changed (1) hide show
  1. src/chains/pipelines.py +7 -1
src/chains/pipelines.py CHANGED
@@ -34,6 +34,7 @@ class PresentationAnalysis(BaseModel):
34
  """Container for presentation analysis results"""
35
  name: str
36
  path: Path
 
37
  metadata: Dict = Field(default_factory=dict)
38
  slides: List[SlideAnalysis] = Field(default_factory=list)
39
  timestamp: str = Field(
@@ -155,6 +156,7 @@ class PresentationPipeline(Chain):
155
  base_path: Base path for storing analysis results
156
  """
157
  super().__init__(**kwargs)
 
158
  self._slide_pipeline = SingleSlidePipeline(
159
  llm=llm,
160
  vision_prompt=vision_prompt,
@@ -252,7 +254,11 @@ class PresentationPipeline(Chain):
252
  presentation = (
253
  PresentationAnalysis.load(latest_analysis)
254
  if latest_analysis
255
- else PresentationAnalysis(name=pdf_path.stem, path=pdf_path)
 
 
 
 
256
  )
257
 
258
  # Get set of already processed pages
 
34
  """Container for presentation analysis results"""
35
  name: str
36
  path: Path
37
+ vision_prompt: str
38
  metadata: Dict = Field(default_factory=dict)
39
  slides: List[SlideAnalysis] = Field(default_factory=list)
40
  timestamp: str = Field(
 
156
  base_path: Base path for storing analysis results
157
  """
158
  super().__init__(**kwargs)
159
+ self._vision_prompt = vision_prompt
160
  self._slide_pipeline = SingleSlidePipeline(
161
  llm=llm,
162
  vision_prompt=vision_prompt,
 
254
  presentation = (
255
  PresentationAnalysis.load(latest_analysis)
256
  if latest_analysis
257
+ else PresentationAnalysis(
258
+ name=pdf_path.stem,
259
+ path=pdf_path,
260
+ vision_prompt=self._vision_prompt
261
+ )
262
  )
263
 
264
  # Get set of already processed pages