togethercomputer / together-python

The Official Python Client for Together's API
https://pypi.org/project/together/
Apache License 2.0
31 stars 7 forks source link

Add Openai compatible Completion class #60

Closed orangetin closed 11 months ago

orangetin commented 11 months ago
orangetin commented 11 months ago

Streaming example with openai:

import openai

response = openai.Completion.create(
    model="davinci-002",
    prompt="hello",
    api_key="sk-xxxx",
    max_tokens=512,
    temperature=0.2,
    stream=True,
)

for tok in response:
    id = tok.id
    delta = tok.choices[0].text
    print(delta, end="", flush=True)
print("\n")

With Together:

import together as openai

response = openai.Completion.create(
    model="togethercomputer/llama-2-70b-chat",
    prompt="hello",
    api_key="xxxx",
    max_tokens=512,
    temperature=0.2,
    stream=True,
)

for tok in response:
    id = tok.id
    delta = tok.choices[0].text
    print(delta, end="", flush=True)
print("\n")

Diff:

--- openai_example.py   2023-11-21 23:17:50
+++ together_example.py 2023-11-21 23:17:56
@@ -1,9 +1,9 @@
-import openai
+import together as openai

 response = openai.Completion.create(
-    model="davinci-002",
+    model="togethercomputer/llama-2-70b-chat",
     prompt="hello",
-    api_key="sk-xxx",
+    api_key="xxxx",
     max_tokens=512,
     temperature=0.2,
     stream=True,