mirror of
https://github.com/xtekky/gpt4free.git
synced 2024-12-24 11:34:02 +03:00
2.0 KiB
2.0 KiB
Create Provider with AI Tool
Call in your terminal the create_provider
script:
python -m etc.tool.create_provider
- Enter your name for the new provider.
- Copy and paste the
cURL
command from your browser developer tools. - Let the AI create the provider for you.
- Customize the provider according to your needs.
Create Provider
- Check out the current list of potential providers, or find your own provider source!
- Create a new file in g4f/Provider with the name of the Provider.
- Implement a class that extends BaseProvider.
from __future__ import annotations
from ..typing import AsyncResult, Messages
from .base_provider import AsyncGeneratorProvider
class HogeService(AsyncGeneratorProvider):
url = "https://chat-gpt.com"
working = True
supports_gpt_35_turbo = True
@classmethod
async def create_async_generator(
cls,
model: str,
messages: Messages,
proxy: str = None,
**kwargs
) -> AsyncResult:
yield ""
- Here, you can adjust the settings, for example, if the website does support streaming, set
supports_stream
toTrue
... - Write code to request the provider in
create_async_generator
andyield
the response, even if it's a one-time response, do not hesitate to look at other providers for inspiration. - Add the Provider Import in
g4f/Provider/__init__.py
from .HogeService import HogeService
__all__ = [
HogeService,
]
- You are done !, test the provider by calling it:
import g4f
response = g4f.ChatCompletion.create(model='gpt-3.5-turbo', provider=g4f.Provider.PROVIDERNAME,
messages=[{"role": "user", "content": "test"}], stream=g4f.Provider.PROVIDERNAME.supports_stream)
for message in response:
print(message, flush=True, end='')