diff options
Diffstat (limited to 'testing/test_interference.py')
-rw-r--r-- | testing/test_interference.py | 27 |
1 files changed, 0 insertions, 27 deletions
diff --git a/testing/test_interference.py b/testing/test_interference.py deleted file mode 100644 index d8e85a6c..00000000 --- a/testing/test_interference.py +++ /dev/null @@ -1,27 +0,0 @@ -# type: ignore -import openai - -openai.api_key = "" -openai.api_base = "http://localhost:1337" - - -def main(): - chat_completion = openai.ChatCompletion.create( - model="gpt-3.5-turbo", - messages=[{"role": "user", "content": "write a poem about a tree"}], - stream=True, - ) - - if isinstance(chat_completion, dict): - # not stream - print(chat_completion.choices[0].message.content) - else: - # stream - for token in chat_completion: - content = token["choices"][0]["delta"].get("content") - if content != None: - print(content, end="", flush=True) - - -if __name__ == "__main__": - main()
\ No newline at end of file |