diff options
| author | Tekky <98614666+xtekky@users.noreply.github.com> | 2023-10-10 16:30:05 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-10 16:30:05 +0200 |
| commit | e06012ae067abf513633134f6bae41bfab2a6497 (patch) | |
| tree | 7b2c4ea4cb468fc046fb4567292d3dfd5d0e7a19 /g4f/Provider/helper.py | |
| parent | ~ | Merge pull request #1033 from Commenter123321/main (diff) | |
| parent | Some small fixes (diff) | |
| download | gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.gz gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.bz2 gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.lz gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.xz gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.tar.zst gpt4free-e06012ae067abf513633134f6bae41bfab2a6497.zip | |
Diffstat (limited to 'g4f/Provider/helper.py')
| -rw-r--r-- | g4f/Provider/helper.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/g4f/Provider/helper.py b/g4f/Provider/helper.py index 5a9a9329..db19adc1 100644 --- a/g4f/Provider/helper.py +++ b/g4f/Provider/helper.py @@ -4,7 +4,7 @@ import asyncio import sys from asyncio import AbstractEventLoop from os import path -from typing import Dict, List +from ..typing import Dict, List, Messages import browser_cookie3 # Change event loop policy on windows @@ -53,7 +53,7 @@ def get_cookies(cookie_domain: str) -> Dict[str, str]: return _cookies[cookie_domain] -def format_prompt(messages: List[Dict[str, str]], add_special_tokens=False) -> str: +def format_prompt(messages: Messages, add_special_tokens=False) -> str: if add_special_tokens or len(messages) > 1: formatted = "\n".join( [ |