github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-28 14:27:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix:core agent dependent prompt service bug (#2129)

Browse Source
This commit is contained in:
Aries-ckt 2024-11-18 09:46:18 +08:00 committed by GitHub
parent 2decf02c12
commit b392d51adf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dbgpt/agent/core/plan/awel/agent_operator.py
View File

@ -22,7 +22,6 @@ from dbgpt.core.interface.message import ModelMessageRoleType
# TODO: Don't dependent on MixinLLMOperator # TODO: Don't dependent on MixinLLMOperator
from dbgpt.model.operators.llm_operator import MixinLLMOperator from dbgpt.model.operators.llm_operator import MixinLLMOperator
from dbgpt.serve.prompt.api.endpoints import get_service
from dbgpt.util.i18n_utils import _ from dbgpt.util.i18n_utils import _
from .... import ActionOutput from .... import ActionOutput
@ -291,6 +290,7 @@ class AWELAgentOperator(
prompt_template = None prompt_template = None
if self.awel_agent.agent_prompt: if self.awel_agent.agent_prompt:
from dbgpt.serve.prompt.api.endpoints import get_service
prompt_service = get_service() prompt_service = get_service()
prompt_template = prompt_service.get_template( prompt_template = prompt_service.get_template(
self.awel_agent.agent_prompt.code self.awel_agent.agent_prompt.code

3
dbgpt/agent/core/plan/awel/agent_operator_resource.py
View File

@ -12,8 +12,6 @@ from dbgpt.core.awel.flow import (
ResourceCategory, ResourceCategory,
register_resource, register_resource,
) )
from dbgpt.serve.prompt.api.endpoints import get_service
from ....resource.base import AgentResource, ResourceType from ....resource.base import AgentResource, ResourceType
from ....resource.manage import get_resource_manager from ....resource.manage import get_resource_manager
from ....util.llm.llm import LLMConfig, LLMStrategyType from ....util.llm.llm import LLMConfig, LLMStrategyType
@ -21,6 +19,7 @@ from ...agent_manage import get_agent_manager
def _agent_resource_prompt_values() -> List[OptionValue]: def _agent_resource_prompt_values() -> List[OptionValue]:
from dbgpt.serve.prompt.api.endpoints import get_service
prompt_service = get_service() prompt_service = get_service()
prompts = prompt_service.get_target_prompt() prompts = prompt_service.get_target_prompt()
return [ return [