aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-18 14:34:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix dict type annotation

Browse Source
This commit is contained in:
youkaichao
2023-04-15 22:10:22 +08:00
parent afd2c5e2c6
commit a51b37f01c
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
autogpt/llm_utils.py
View File

@@ -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.

6
autogpt/processing/text.py
View File

@@ -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",