Update openai_requests.py
Browse files- openai_requests.py +39 -0
openai_requests.py
CHANGED
@@ -56,6 +56,45 @@ def process_transcribation_with_assistant(prompt, transcript):
|
|
56 |
},
|
57 |
headers=headers,
|
58 |
stream=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
59 |
|
60 |
|
61 |
|
|
|
56 |
},
|
57 |
headers=headers,
|
58 |
stream=True)
|
59 |
+
|
60 |
+
text_container = st.empty()
|
61 |
+
output_text = ''
|
62 |
+
event_name = ""
|
63 |
+
for line in run_response.iter_lines(decode_unicode=True):
|
64 |
+
if line:
|
65 |
+
if line.startswith("event:"):
|
66 |
+
event_name = line.split(":")[1].strip()
|
67 |
+
if event_name == 'done':
|
68 |
+
break
|
69 |
+
elif line.startswith("data:") and event_name == 'thread.message.delta':
|
70 |
+
event_data = json.loads(line.split(":", 1)[1].strip())
|
71 |
+
output_text += event_data['delta']['content'][0]['text']['value']
|
72 |
+
text_container.text(output_text)
|
73 |
+
|
74 |
+
for i in range(number_of_runs - 1):
|
75 |
+
message_response = requests.post(baseUrl + '/threads/' + thread_id + '/messages',
|
76 |
+
json={"role" : "user", "content": "Продолжай работать на текущей задачей"},
|
77 |
+
headers=headers)
|
78 |
+
|
79 |
+
run_response = requests.post(baseUrl + '/threads/' + thread_id + '/runs',
|
80 |
+
json={
|
81 |
+
"assistant_id": st.secrets["OPENAI_ASSISTANT_ID"],
|
82 |
+
"stream": True
|
83 |
+
},
|
84 |
+
headers=headers,
|
85 |
+
stream=True)
|
86 |
+
event_name = ""
|
87 |
+
response = ''
|
88 |
+
for line in run_response.iter_lines(decode_unicode=True):
|
89 |
+
if line:
|
90 |
+
if line.startswith("event:"):
|
91 |
+
event_name = line.split(":")[1].strip()
|
92 |
+
if event_name == 'done':
|
93 |
+
break
|
94 |
+
elif line.startswith("data:") and event_name == 'thread.message.delta':
|
95 |
+
event_data = json.loads(line.split(":", 1)[1].strip())
|
96 |
+
output_text += event_data['delta']['content'][0]['text']['value']
|
97 |
+
text_container.text(output_text)
|
98 |
|
99 |
|
100 |
|