Merge branch 'langgenius:main' into main

langgenius · Apr 10, 2024 · 7a8873d · 7a8873d
2 parents 2083fc3 + 0f897bc
commit 7a8873d
Show file tree

Hide file tree

Showing 47 changed files with 3,295 additions and 1,089 deletions.
diff --git a/README.md b/README.md
diff --git a/api/core/app/apps/chat/app_runner.py b/api/core/app/apps/chat/app_runner.py
@@ -156,6 +156,8 @@ def run(self, application_generate_entity: ChatAppGenerateEntity,
 
             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,

diff --git a/api/core/app/apps/completion/app_runner.py b/api/core/app/apps/completion/app_runner.py
@@ -116,6 +116,8 @@ def run(self, application_generate_entity: CompletionAppGenerateEntity,
 
             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,

diff --git a/api/core/rag/retrieval/agent/fake_llm.py b/api/core/rag/retrieval/agent/fake_llm.py
diff --git a/api/core/rag/retrieval/agent/llm_chain.py b/api/core/rag/retrieval/agent/llm_chain.py
diff --git a/api/core/rag/retrieval/agent/multi_dataset_router_agent.py b/api/core/rag/retrieval/agent/multi_dataset_router_agent.py
diff --git a/api/core/rag/retrieval/agent/output_parser/__init__.py b/api/core/rag/retrieval/agent/output_parser/__init__.py