Former-commit-id: e479c53ed7
group-chat
Kye 1 year ago
parent f31c241b69
commit 1764a738e3

@ -51,9 +51,9 @@ class Worker:
``` ```
""" """
@log_decorator # @log_decorator
@error_decorator # @error_decorator
@timing_decorator # @timing_decorator
def __init__( def __init__(
self, self,
model_name="gpt-4", model_name="gpt-4",
@ -140,9 +140,9 @@ class Worker:
# return results # return results
@log_decorator # @log_decorator
@error_decorator # @error_decorator
@timing_decorator # @timing_decorator
def setup_tools(self, external_tools): def setup_tools(self, external_tools):
""" """
Set up tools for the worker. Set up tools for the worker.
@ -205,9 +205,9 @@ class Worker:
except Exception as error: except Exception as error:
raise RuntimeError(f"Error setting up agent: {error}") raise RuntimeError(f"Error setting up agent: {error}")
@log_decorator # @log_decorator
@error_decorator # @error_decorator
@timing_decorator # @timing_decorator
def run(self, task): def run(self, task):
""" """
Run the autonomous agent on a given task. Run the autonomous agent on a given task.
@ -224,9 +224,9 @@ class Worker:
except Exception as error: except Exception as error:
raise RuntimeError(f"Error while running agent: {error}") raise RuntimeError(f"Error while running agent: {error}")
@log_decorator # @log_decorator
@error_decorator # @error_decorator
@timing_decorator # @timing_decorator
def __call__(self, task): def __call__(self, task):
""" """
Make the worker callable to run the agent on a given task. Make the worker callable to run the agent on a given task.

Loading…
Cancel
Save