Skip to content

Commit

Permalink
Remove tenacity retry decorators from
Browse files Browse the repository at this point in the history
RoleAssignmentAgent methods
  • Loading branch information
Appointat committed Nov 12, 2023
1 parent a12af53 commit f50ba8b
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions camel/agents/role_assignment_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,6 @@

import matplotlib.pyplot as plt
import networkx as nx
from tenacity import retry, stop_after_attempt, wait_exponential

from camel.agents import ChatAgent
from camel.agents.insight_agent import InsightAgent
Expand Down Expand Up @@ -50,7 +49,6 @@ def __init__(
super().__init__(system_message, model, model_config)
self.model_config = model_config or ChatGPTConfig()

@retry(wait=wait_exponential(min=5, max=60), stop=stop_after_attempt(5))
def run_role_with_description(
self,
task_prompt: Union[str, TextPrompt],
Expand Down Expand Up @@ -263,7 +261,6 @@ def sort_oriented_graph(

return parallel_subtask_pipelines

@retry(wait=wait_exponential(min=5, max=60), stop=stop_after_attempt(5))
def split_tasks(
self,
task_prompt: Union[str, TextPrompt],
Expand Down Expand Up @@ -454,7 +451,6 @@ def split_tasks(

return subtasks_with_dependencies_dict

@retry(wait=wait_exponential(min=5, max=60), stop=stop_after_attempt(5))
def evaluate_role_compatibility(
self,
subtask_prompt: Union[str, TextPrompt],
Expand Down Expand Up @@ -556,7 +552,6 @@ def evaluate_role_compatibility(

return role_compatibility_scores_dict

@retry(wait=wait_exponential(min=5, max=60), stop=stop_after_attempt(5))
def get_retrieval_index_from_environment(
self,
labels_sets: List[List[str]],
Expand Down Expand Up @@ -655,7 +650,6 @@ def get_retrieval_index_from_environment(
return target_labels_indices, labels_sets_indices, \
target_retrieved_labels, labels_retrieved_sets

@retry(wait=wait_exponential(min=5, max=60), stop=stop_after_attempt(5))
def transform_dialogue_into_text(
self, user: str, assistant: str, task_prompt: str,
user_conversation: str,
Expand Down

0 comments on commit f50ba8b

Please sign in to comment.