mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-12-06 02:30:41 -08:00
Fix api and None provider
This commit is contained in:
parent
74a21cdb4e
commit
3a81f9a2af
2 changed files with 39 additions and 20 deletions
27
etc/testing/test_api.py
Normal file
27
etc/testing/test_api.py
Normal file
|
|
@ -0,0 +1,27 @@
|
|||
import openai
|
||||
|
||||
# Set your Hugging Face token as the API key if you use embeddings
|
||||
# If you don't use embeddings, leave it empty
|
||||
openai.api_key = "YOUR_HUGGING_FACE_TOKEN" # Replace with your actual token
|
||||
|
||||
# Set the API base URL if needed, e.g., for a local development environment
|
||||
openai.api_base = "http://localhost:1337/v1"
|
||||
|
||||
def main():
|
||||
response = openai.ChatCompletion.create(
|
||||
model="gpt-3.5-turbo",
|
||||
messages=[{"role": "user", "content": "write a poem about a tree"}],
|
||||
stream=True,
|
||||
)
|
||||
if isinstance(response, dict):
|
||||
# Not streaming
|
||||
print(response.choices[0].message.content)
|
||||
else:
|
||||
# Streaming
|
||||
for token in response:
|
||||
content = token["choices"][0]["delta"].get("content")
|
||||
if content is not None:
|
||||
print(content, end="", flush=True)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue