Spaces:
Sleeping
Sleeping
File size: 586 Bytes
22b1ec3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from llama_index.llms.perplexity import Perplexity
from llama_index.core.llms import ChatMessage
def get_response_from_perplexity(question: str, perplexity_key):
# Initialize Perplexity LLM
perplexity_llm = Perplexity(
api_key=perplexity_key,
model="sonar",
temperature=0.2
)
messages_dict = [
{"role": "system", "content": "answer in 2-3 sentences"},
{"role": "user", "content": question},
]
messages = [ChatMessage(**msg) for msg in messages_dict]
response = perplexity_llm.chat(messages)
return response
|