From a51b37f01c68fc00308ab01a47e6c93e34906fb3 Mon Sep 17 00:00:00 2001 From: youkaichao Date: Sat, 15 Apr 2023 22:10:22 +0800 Subject: [PATCH] fix dict type annotation --- autogpt/llm_utils.py | 2 +- autogpt/processing/text.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/autogpt/llm_utils.py b/autogpt/llm_utils.py index f2418b65..260403e2 100644 --- a/autogpt/llm_utils.py +++ b/autogpt/llm_utils.py @@ -59,7 +59,7 @@ def create_chat_completion( """Create a chat completion using the OpenAI API Args: - messages (list[dict[str, str]]): The messages to send to the chat completion + messages (List[Dict[str, str]]): The messages to send to the chat completion model (str, optional): The model to use. Defaults to None. temperature (float, optional): The temperature to use. Defaults to 0.9. max_tokens (int, optional): The max tokens to use. Defaults to None. diff --git a/autogpt/processing/text.py b/autogpt/processing/text.py index 24e3bb85..d9e335f8 100644 --- a/autogpt/processing/text.py +++ b/autogpt/processing/text.py @@ -1,5 +1,5 @@ """Text processing functions""" -from typing import Generator, Optional +from typing import Generator, Optional, Dict from selenium.webdriver.remote.webdriver import WebDriver from autogpt.memory import get_memory from autogpt.config import Config @@ -114,7 +114,7 @@ def scroll_to_percentage(driver: WebDriver, ratio: float) -> None: driver.execute_script(f"window.scrollTo(0, document.body.scrollHeight * {ratio});") -def create_message(chunk: str, question: str) -> dict[str, str]: +def create_message(chunk: str, question: str) -> Dict[str, str]: """Create a message for the chat completion Args: @@ -122,7 +122,7 @@ def create_message(chunk: str, question: str) -> dict[str, str]: question (str): The question to answer Returns: - dict[str, str]: The message to send to the chat completion + Dict[str, str]: The message to send to the chat completion """ return { "role": "user",