summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorH Lohaus <hlohaus@users.noreply.github.com>2024-01-04 20:30:43 +0100
committerGitHub <noreply@github.com>2024-01-04 20:30:43 +0100
commit4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23 (patch)
tree5666afdfd278b8b376c755b325369667dcdb7f41
parentMerge pull request #1430 from ramonvc/main (diff)
parentUpdate openai example (diff)
downloadgpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.tar
gpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.tar.gz
gpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.tar.bz2
gpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.tar.lz
gpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.tar.xz
gpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.tar.zst
gpt4free-4a55337e9e3f14879bdbe4aaaf5b83b4f6840e23.zip
Diffstat (limited to '')
-rw-r--r--README.md19
1 files changed, 11 insertions, 8 deletions
diff --git a/README.md b/README.md
index 15ed1890..6d8f39c9 100644
--- a/README.md
+++ b/README.md
@@ -369,17 +369,19 @@ python -m g4f.api.run
```
```python
-import openai
+from openai 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
+client = OpenAI(
+ # Set your Hugging Face token as the API key if you use embeddings
+ api_key="YOUR_HUGGING_FACE_TOKEN",
+
+ # Set the API base URL if needed, e.g., for a local development environment
+ base_url="http://localhost:1337/v1"
+)
-# Set the API base URL if needed, e.g., for a local development environment
-openai.api_base = "http://localhost:1337/v1"
def main():
- chat_completion = openai.ChatCompletion.create(
+ chat_completion = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "write a poem about a tree"}],
stream=True,
@@ -391,10 +393,11 @@ def main():
else:
# Streaming
for token in chat_completion:
- content = token["choices"][0]["delta"].get("content")
+ content = token.choices[0].delta.content
if content is not None:
print(content, end="", flush=True)
+
if __name__ == "__main__":
main()
```