Baskar2005 commited on
Commit
1bd26fe
·
verified ·
1 Parent(s): 604e6b8

Update dspy_qa.py

Browse files
Files changed (1) hide show
  1. dspy_qa.py +2 -2
dspy_qa.py CHANGED
@@ -32,8 +32,8 @@ class DocQA(dspy.Module):
32
  def __init__(self, file_path,passages_per_hop=3, max_hops=2):
33
  super().__init__()
34
  self.cache = "cache.json"
35
- self.llm = dspy.AzureOpenAI(api_base=os.getenv("AZURE_OPENAI_ENDPOINT"),
36
- api_version=os.getenv("OPENAI_API_VERSION"),
37
  model="GPT-4o")
38
 
39
  self.generate_query = [dspy.ChainOfThought(GenerateSearchQuery) for _ in range(max_hops)]
 
32
  def __init__(self, file_path,passages_per_hop=3, max_hops=2):
33
  super().__init__()
34
  self.cache = "cache.json"
35
+ self.llm = dspy.AzureOpenAI(api_base="https://eastus2.api.cognitive.microsoft.com/",
36
+ api_version="2024-02-01",
37
  model="GPT-4o")
38
 
39
  self.generate_query = [dspy.ChainOfThought(GenerateSearchQuery) for _ in range(max_hops)]