Skip to content

Commit

Permalink
Fix format
Browse files Browse the repository at this point in the history
  • Loading branch information
thinkall committed Oct 16, 2023
1 parent 2d2e3af commit 3ced5cd
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions autogen/agentchat/contrib/retrieve_user_proxy_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def __init__(
The function should take a string as input and return three integers (token_count, tokens_per_message, tokens_per_name).
Default is None, tiktoken will be used and may not be accurate for non-OpenAI models.
- n_results (Optional, int): the number of results to be retrieved. Useful in group chat. Will be overridden by the same
parameter passed to `generate_init_message`. Default is 20.
parameter passed to `generate_init_message`. Default is 20.
**kwargs (dict): other kwargs in [UserProxyAgent](../user_proxy_agent#__init__).
Example of overriding retrieve_docs:
Expand Down Expand Up @@ -397,7 +397,7 @@ def generate_init_message(self, problem: str, n_results: int = None, search_stri
Args:
problem (str): the problem to be solved.
n_results (int): the number of results to be retrieved. Default is None, will use the value set in retrieve_config.
n_results (int): the number of results to be retrieved. Default is None, will use the value set in retrieve_config.
search_string (str): only docs containing this string will be retrieved.
Returns:
Expand Down

0 comments on commit 3ced5cd

Please sign in to comment.