Get suggested query to build on top of a given query or as a similar query
(scope, failed_query, parent_id=None, session_id=None)
| 117 | |
| 118 | |
| 119 | def generate_suggestion(scope, failed_query, parent_id=None, session_id=None): |
| 120 | """ |
| 121 | Get suggested query to build on top of a given query or as a similar query |
| 122 | """ |
| 123 | messages = _get_query_suggestion_messages(scope) |
| 124 | |
| 125 | prompt = _get_query_suggestion_message(scope, failed_query) |
| 126 | model = "gpt-3.5-turbo" |
| 127 | |
| 128 | messages.append({ |
| 129 | "role": "user", |
| 130 | "content": prompt |
| 131 | }) |
| 132 | response = get_assistant_message_from_openai( |
| 133 | messages=messages, |
| 134 | model=model, |
| 135 | scope="USA", |
| 136 | purpose="query_suggestion" |
| 137 | )["message"]["content"] |
| 138 | suggested_query = response |
| 139 | |
| 140 | suggestion_id = log_suggested_query( |
| 141 | input_text=failed_query, |
| 142 | reason="successful_query_suggestion", |
| 143 | parent_id=parent_id, |
| 144 | suggested_query=suggested_query, |
| 145 | app_name=scope, |
| 146 | prompt=prompt, |
| 147 | model=model, |
| 148 | session_id=session_id |
| 149 | ) |
| 150 | |
| 151 | return suggested_query, str(suggestion_id) |
no test coverage detected