28 lines
1000 B
Python
28 lines
1000 B
Python
import sys
|
|
import openai
|
|
import json
|
|
|
|
def main(user_message):
|
|
# Set your OpenAI API key from the environment, or replace with your key here
|
|
openai.api_key = "sk-proj-CMVUSsmXIr-Da3a8bpAByG0v2FD1hxEahGs7CqTz7tcegAWGP03OitAAD5LQVf_z5ZUucDWZ10pSHXJVzoWZeGCHueskkC5IMLccUldlvTlsfUA"
|
|
|
|
try:
|
|
# Call the OpenAI ChatCompletion API
|
|
response = openai.ChatCompletion.create(
|
|
model="gpt-3.5-turbo",
|
|
messages=[{"role": "user", "content": user_message}],
|
|
assistant_id="asst_0LyJEWQS90Sx0oEIYBVWE47C" # Assistant ID
|
|
)
|
|
|
|
assistant_message = response['choices'][0]['message']['content']
|
|
print(json.dumps({"response": assistant_message}))
|
|
|
|
except Exception as e:
|
|
print(json.dumps({"response": f"An error occurred: {str(e)}"}))
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
user_message = sys.argv[1]
|
|
main(user_message)
|
|
else:
|
|
print(json.dumps({"response": "No message provided"})) |