Merge branch 'feat/r2' into deploy/rag-dev

feat/datasource
jyong 10 months ago
commit f4e1ea9011

@ -245,7 +245,9 @@ class GetOnlineDocumentPageContentRequest(BaseModel):
Get online document page content request Get online document page content request
""" """
online_document_info: OnlineDocumentInfo workspace_id: str = Field(..., description="The workspace id")
page_id: str = Field(..., description="The page id")
type: str = Field(..., description="The type of the page")
class OnlineDocumentPageContent(BaseModel): class OnlineDocumentPageContent(BaseModel):

@ -180,7 +180,7 @@ class PluginDatasourceManager(BasePluginClient):
"provider": datasource_provider_id.provider_name, "provider": datasource_provider_id.provider_name,
"datasource": datasource_name, "datasource": datasource_name,
"credentials": credentials, "credentials": credentials,
"datasource_parameters": datasource_parameters, "page": datasource_parameters,
}, },
}, },
headers={ headers={

@ -103,7 +103,11 @@ class DatasourceNode(BaseNode[DatasourceNodeData]):
online_document_result: Generator[DatasourceMessage, None, None] = ( online_document_result: Generator[DatasourceMessage, None, None] = (
datasource_runtime.get_online_document_page_content( datasource_runtime.get_online_document_page_content(
user_id=self.user_id, user_id=self.user_id,
datasource_parameters=GetOnlineDocumentPageContentRequest(**parameters), datasource_parameters=GetOnlineDocumentPageContentRequest(
workspace_id=datasource_info.get("workspace_id"),
page_id=datasource_info.get("page").get("page_id"),
type=datasource_info.get("type"),
),
provider_type=datasource_type, provider_type=datasource_type,
) )
) )

Loading…
Cancel
Save