Spaces:
Sleeping
Sleeping
Update feifeilib/feifeichat.py
Browse files- feifeilib/feifeichat.py +2 -2
feifeilib/feifeichat.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
from openai import OpenAI
|
| 3 |
|
| 4 |
-
|
| 5 |
def feifeiprompt(message_text=""):
|
| 6 |
input_prompt = []
|
| 7 |
feifei_photo = "You are a helpful assistant."
|
|
@@ -18,7 +18,7 @@ def feifeiprompt(message_text=""):
|
|
| 18 |
def feifeichat(feifei_chat_text1="", feifei_chat_text2=""):
|
| 19 |
message_text = f"{feifei_chat_text1}, {feifei_chat_text2},Create an English photo description based on the given content"
|
| 20 |
feifei_prompt = feifeiprompt(message_text)
|
| 21 |
-
completion =
|
| 22 |
model=os.getenv("mymodel"),
|
| 23 |
messages=feifei_prompt,
|
| 24 |
stream=True
|
|
|
|
| 1 |
import os
|
| 2 |
from openai import OpenAI
|
| 3 |
|
| 4 |
+
myclient = OpenAI(api_key=os.getenv("myapikey"), base_url=os.getenv("mybase_url"))
|
| 5 |
def feifeiprompt(message_text=""):
|
| 6 |
input_prompt = []
|
| 7 |
feifei_photo = "You are a helpful assistant."
|
|
|
|
| 18 |
def feifeichat(feifei_chat_text1="", feifei_chat_text2=""):
|
| 19 |
message_text = f"{feifei_chat_text1}, {feifei_chat_text2},Create an English photo description based on the given content"
|
| 20 |
feifei_prompt = feifeiprompt(message_text)
|
| 21 |
+
completion = myclient.chat.completions.create(
|
| 22 |
model=os.getenv("mymodel"),
|
| 23 |
messages=feifei_prompt,
|
| 24 |
stream=True
|