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)
|