🍌🐒 commited on
Commit
5fb84cd
·
1 Parent(s): f53a43f

Fix: Update Perplexity package and client implementation

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -1
  2. src/agents.py +3 -3
requirements.txt CHANGED
@@ -23,4 +23,4 @@ anthropic>=0.18.1
23
  duckduckgo-search>=4.4.3
24
  orjson>=3.9.15
25
  python-dotenv>=1.0.0
26
- git+https://github.com/perplexity-ai/perplexity-python.git
 
23
  duckduckgo-search>=4.4.3
24
  orjson>=3.9.15
25
  python-dotenv>=1.0.0
26
+ perplexity>=0.3.0
src/agents.py CHANGED
@@ -3,7 +3,7 @@ from typing import Optional
3
  import openai
4
  import anthropic
5
  from duckduckgo_search import DDGS
6
- from perplexity import Client
7
 
8
  class Agent:
9
  def __init__(self, base_model: str = "gpt-3.5-turbo", search_engine: str = "duckduckgo"):
@@ -16,7 +16,7 @@ class Agent:
16
  elif "claude" in base_model:
17
  self.claude = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
18
  elif "sonar" in base_model:
19
- self.perplexity = Client(api_key=os.getenv("PERPLEXITY_API_KEY"))
20
 
21
  def execute(self, message: str, project_name: str) -> Optional[str]:
22
  try:
@@ -72,7 +72,7 @@ class Agent:
72
  # Use the mapped model or default to sonar
73
  model = model_mapping.get(self.base_model, "sonar")
74
 
75
- response = self.perplexity.chat.create_completion(
76
  model=model,
77
  messages=[{"role": "user", "content": message}]
78
  )
 
3
  import openai
4
  import anthropic
5
  from duckduckgo_search import DDGS
6
+ from perplexity.client import PerplexityClient
7
 
8
  class Agent:
9
  def __init__(self, base_model: str = "gpt-3.5-turbo", search_engine: str = "duckduckgo"):
 
16
  elif "claude" in base_model:
17
  self.claude = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY"))
18
  elif "sonar" in base_model:
19
+ self.perplexity = PerplexityClient(api_key=os.getenv("PERPLEXITY_API_KEY"))
20
 
21
  def execute(self, message: str, project_name: str) -> Optional[str]:
22
  try:
 
72
  # Use the mapped model or default to sonar
73
  model = model_mapping.get(self.base_model, "sonar")
74
 
75
+ response = self.perplexity.chat.create(
76
  model=model,
77
  messages=[{"role": "user", "content": message}]
78
  )