From 7557fa58d6fe6de7861ba7244fda82f0b4035886 Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Wed, 20 Dec 2023 07:31:37 +0800 Subject: [PATCH 1/3] Update plugins/flytekit-bigquery/flytekitplugins/bigquery/agent.py Co-authored-by: Kevin Su --- plugins/flytekit-bigquery/flytekitplugins/bigquery/agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/plugins/flytekit-bigquery/flytekitplugins/bigquery/agent.py b/plugins/flytekit-bigquery/flytekitplugins/bigquery/agent.py index dfd7d2001f7..84229e3eb29 100644 --- a/plugins/flytekit-bigquery/flytekitplugins/bigquery/agent.py +++ b/plugins/flytekit-bigquery/flytekitplugins/bigquery/agent.py @@ -44,7 +44,7 @@ class Metadata: class BigQueryAgent(AgentBase): def __init__(self): - super().__init__(task_type="bigquery_query_job_task", name="bigquery_query", asynchronous=False) + super().__init__(task_type="bigquery_query_job_task", name="bigquery_agent", asynchronous=False) def create( self, From efc8c427259fb17a56e44ea27d7df574b19ad155 Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Wed, 20 Dec 2023 07:31:49 +0800 Subject: [PATCH 2/3] Update flytekit/extend/backend/base_agent.py Co-authored-by: Kevin Su --- flytekit/extend/backend/base_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flytekit/extend/backend/base_agent.py b/flytekit/extend/backend/base_agent.py index 841c6a7abe9..4002ce43421 100644 --- a/flytekit/extend/backend/base_agent.py +++ b/flytekit/extend/backend/base_agent.py @@ -69,7 +69,7 @@ def task_type(self) -> str: @property def name(self) -> str: """ - task_type is the name of the task type that this agent supports. + The name of the agent. """ return self._name From 703e7bc75db07f71150a40c42dda16359dd2c0ba Mon Sep 17 00:00:00 2001 From: Future-Outlier Date: Wed, 20 Dec 2023 07:31:55 +0800 Subject: [PATCH 3/3] Update flytekit/extend/backend/base_agent.py Co-authored-by: Kevin Su --- flytekit/extend/backend/base_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flytekit/extend/backend/base_agent.py b/flytekit/extend/backend/base_agent.py index 4002ce43421..88443259e2e 100644 --- a/flytekit/extend/backend/base_agent.py +++ b/flytekit/extend/backend/base_agent.py @@ -124,7 +124,7 @@ async def async_delete(self, context: grpc.ServicerContext, resource_meta: bytes class AgentRegistry(object): """ This is the registry for all agents. - The agent service will look up the agent based on the task type. + The agent service will look up the agent registry based on the task type. The agent metadata service will look up the agent metadata based on the agent name. """