pvanand commited on
Commit
d97f279
·
1 Parent(s): 0bc6b7f

Update actions/actions.py

Browse files
Files changed (1) hide show
  1. 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)