diff options
author | Nicholas Tindle <nicholas.tindle@agpt.co> | 2024-05-09 12:59:45 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-05-09 12:59:45 +0200 |
commit | 26b582c0d4fc9b916cdf06d50c1971d195c4f559 (patch) | |
tree | 087e7624cd39f433b3eaa6b2d7f9ecb0242bf9bc /autogpts/autogpt/autogpt/app/agent_protocol_server.py | |
parent | Merge branch 'master' into forge/fixes (diff) | |
parent | fix(agent): Fix type propagation of `Command` and `@command` when used on met... (diff) | |
download | Auto-GPT-forge/fixes.tar.gz Auto-GPT-forge/fixes.tar.bz2 Auto-GPT-forge/fixes.zip |
Merge branch 'master' into forge/fixesforge/fixes
Diffstat (limited to 'autogpts/autogpt/autogpt/app/agent_protocol_server.py')
-rw-r--r-- | autogpts/autogpt/autogpt/app/agent_protocol_server.py | 25 |
1 files changed, 11 insertions, 14 deletions
diff --git a/autogpts/autogpt/autogpt/app/agent_protocol_server.py b/autogpts/autogpt/autogpt/app/agent_protocol_server.py index cdaf1f460..2eb09706e 100644 --- a/autogpts/autogpt/autogpt/app/agent_protocol_server.py +++ b/autogpts/autogpt/autogpt/app/agent_protocol_server.py @@ -34,7 +34,6 @@ from autogpt.agent_manager import AgentManager from autogpt.app.utils import is_port_free from autogpt.config import Config from autogpt.core.resource.model_providers import ChatModelProvider, ModelProviderBudget -from autogpt.core.resource.model_providers.openai import OpenAIProvider from autogpt.file_storage import FileStorage from autogpt.models.action_history import ActionErrorResult, ActionSuccessResult from autogpt.utils.exceptions import AgentFinished @@ -464,20 +463,18 @@ class AgentProtocolServer: if task.additional_input and (user_id := task.additional_input.get("user_id")): _extra_request_headers["AutoGPT-UserID"] = user_id - task_llm_provider = None - if isinstance(self.llm_provider, OpenAIProvider): - settings = self.llm_provider._settings.copy() - settings.budget = task_llm_budget - settings.configuration = task_llm_provider_config # type: ignore - task_llm_provider = OpenAIProvider( - settings=settings, - logger=logger.getChild(f"Task-{task.task_id}_OpenAIProvider"), - ) - - if task_llm_provider and task_llm_provider._budget: - self._task_budgets[task.task_id] = task_llm_provider._budget + settings = self.llm_provider._settings.copy() + settings.budget = task_llm_budget + settings.configuration = task_llm_provider_config + task_llm_provider = self.llm_provider.__class__( + settings=settings, + logger=logger.getChild( + f"Task-{task.task_id}_{self.llm_provider.__class__.__name__}" + ), + ) + self._task_budgets[task.task_id] = task_llm_provider._budget # type: ignore - return task_llm_provider or self.llm_provider + return task_llm_provider def task_agent_id(task_id: str | int) -> str: |