(prompt, token=2500)
| 13 | |
| 14 | @retry(wait=wait_random_exponential(min=1, max=20), stop=stop_after_attempt(6)) |
| 15 | def generate_response(prompt, token=2500): |
| 16 | response = openai.Completion.create( |
| 17 | model='text-davinci-003', |
| 18 | prompt=prompt, |
| 19 | temperature=0.1, |
| 20 | max_tokens=min(2500,token), |
| 21 | ) |
| 22 | return response['choices'][0]['text'] |
| 23 | |
| 24 | @retry(wait=wait_random_exponential(min=1, max=20), stop=stop_after_attempt(6)) |
| 25 | def get_embedding(text: str, model="text-embedding-ada-002"): |
no outgoing calls
no test coverage detected