| from control import recommendation_handler | |
| import numpy as np | |
| prompt_json, _ = recommendation_handler.populate_json() | |
| prompt = """ | |
| Hello how are you? | |
| . """ | |
| model_id='sentence-transformers/all-MiniLM-L6-v2' | |
| embedding_fn = recommendation_handler.get_embedding_func(inference='local', model_id=model_id) | |
| print(recommendation_handler.split_into_sentences(prompt)) | |
| recommendation_json = recommendation_handler.recommend_prompt(prompt, prompt_json, embedding_fn=embedding_fn, get_xy=True) | |
| print(recommendation_json) |