File size: 629 Bytes
132ac81 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from functools import lru_cache
from requests import Response, post
from constants import MODEL_NAME, MODEL_URL
def build_body(prompt: str) -> dict:
return {"model": MODEL_NAME, "prompt": prompt, "stream": False}
@lru_cache
def ask_assistant(field_name: str) -> str:
prompt: str = """
Please, provide the
AWS Role:
AWS Path :
Responsablities:
Description:
based on the current employee role below.
Role: {input}
"""
response: Response = post(MODEL_URL, json=build_body(prompt.format(input=field_name)))
response_txt: str = response.json()["response"]
return response_txt
|