youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(datasource): change datasource result type to event-stream

Browse Source
This commit is contained in:
Dongyu Li
2025-06-19 11:10:24 +08:00
parent 02ae479636
commit 82d0a70cb4
4 changed files with 115 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/rag/entities/event.py
View File

@ -11,11 +11,16 @@ class DatasourceStreamEvent(Enum):
"""
PROCESSING = "datasource_processing"
COMPLETED = "datasource_completed"
ERROR = "datasource_error"
class BaseDatasourceEvent(BaseModel):
pass
class DatasourceErrorEvent(BaseDatasourceEvent):
event: str = DatasourceStreamEvent.ERROR.value
error: str = Field(..., description="error message")
class DatasourceCompletedEvent(BaseDatasourceEvent):
event: str = DatasourceStreamEvent.COMPLETED.value
data: Mapping[str,Any] | list = Field(..., description="result")