src/assistant.py

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