Spaces:
Sleeping
Sleeping
Update actions/actions.py
Browse files- actions/actions.py +30 -0
actions/actions.py
CHANGED
@@ -152,6 +152,36 @@ class SayHelloWorld(Action):
|
|
152 |
stop=None,
|
153 |
)
|
154 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
155 |
# Output the generated response to user
|
156 |
generated_text = response.choices[0].text
|
157 |
dispatcher.utter_message(text=generated_text)
|
|
|
152 |
stop=None,
|
153 |
)
|
154 |
|
155 |
+
# Output the generated response to user
|
156 |
+
generated_text = response.choices[0].text
|
157 |
+
dispatcher.utter_message(text=generated_text)
|
158 |
+
|
159 |
+
|
160 |
+
class GetOpenAIResponse(Action):
|
161 |
+
|
162 |
+
def name(self) -> Text:
|
163 |
+
return "action_get_response_openai"
|
164 |
+
|
165 |
+
def run(self,
|
166 |
+
dispatcher: CollectingDispatcher,
|
167 |
+
tracker: Tracker,
|
168 |
+
domain: Dict[Text, Any]) -> List[Dict[Text, Any]]:
|
169 |
+
|
170 |
+
# Use OpenAI API to generate a response
|
171 |
+
secret_value_0 = os.environ.get("openai")
|
172 |
+
openai.api_key = secret_value_0
|
173 |
+
model_engine = "text-davinci-002"
|
174 |
+
prompt_template = tracker.latest_message.get('text')
|
175 |
+
|
176 |
+
response = openai.Completion.create(
|
177 |
+
engine=model_engine,
|
178 |
+
prompt=prompt_template,
|
179 |
+
max_tokens=124,
|
180 |
+
temperature=0.8,
|
181 |
+
n=1,
|
182 |
+
stop=None,
|
183 |
+
)
|
184 |
+
|
185 |
# Output the generated response to user
|
186 |
generated_text = response.choices[0].text
|
187 |
dispatcher.utter_message(text=generated_text)
|