Priya-0914 commited on
Commit
22b1ec3
·
verified ·
1 Parent(s): 2b371d5

Create perplexity.py

Browse files
Files changed (1) hide show
  1. perplexity.py +22 -0
perplexity.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from llama_index.llms.perplexity import Perplexity
2
+ from llama_index.core.llms import ChatMessage
3
+
4
+ def get_response_from_perplexity(question: str, perplexity_key):
5
+
6
+ # Initialize Perplexity LLM
7
+ perplexity_llm = Perplexity(
8
+ api_key=perplexity_key,
9
+ model="sonar",
10
+ temperature=0.2
11
+ )
12
+ messages_dict = [
13
+ {"role": "system", "content": "answer in 2-3 sentences"},
14
+ {"role": "user", "content": question},
15
+ ]
16
+ messages = [ChatMessage(**msg) for msg in messages_dict]
17
+ response = perplexity_llm.chat(messages)
18
+ return response
19
+
20
+
21
+
22
+