Spaces:
Build error
Build error
Ilia Tambovtsev commited on
Commit ·
b7dc0e7
1
Parent(s): 4c785d2
fix: implement resetting prompt
Browse files- 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
|
| 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)}")
|