hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 03:16:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove langchain dataset retrival agent logic (#3311)

Browse Source
This commit is contained in:
Jyong
2024-04-10 20:37:22 +08:00
committed by GitHub
parent 8cefa6b82e
commit b6de97ad53
14 changed files with 341 additions and 966 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/app/apps/chat/app_runner.py
View File

@@ -156,6 +156,8 @@ class ChatAppRunner(AppRunner):
dataset_retrieval = DatasetRetrieval()
context = dataset_retrieval.retrieve(
app_id=app_record.id,
user_id=application_generate_entity.user_id,
tenant_id=app_record.tenant_id,
model_config=application_generate_entity.model_config,
config=app_config.dataset,

2
api/core/app/apps/completion/app_runner.py
View File

@@ -116,6 +116,8 @@ class CompletionAppRunner(AppRunner):
dataset_retrieval = DatasetRetrieval()
context = dataset_retrieval.retrieve(
app_id=app_record.id,
user_id=application_generate_entity.user_id,
tenant_id=app_record.tenant_id,
model_config=application_generate_entity.model_config,
config=dataset_config,