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

Update dspy_qa.py

Browse files
Files changed (1) hide show
  1. dspy_qa.py +3 -4
dspy_qa.py CHANGED
@@ -6,8 +6,7 @@ from dspy.retrieve.faiss_rm import FaissRM
6
  from langchain_community.document_loaders import PyPDFLoader
7
  from langchain_text_splitters import RecursiveCharacterTextSplitter
8
 
9
- api_base=os.getenv("AZURE_OPENAI_ENDPOINT")
10
- api_version=os.getenv("OPENAI_API_VERSION")
11
 
12
 
13
 
@@ -33,8 +32,8 @@ class DocQA(dspy.Module):
33
  def __init__(self, file_path,passages_per_hop=3, max_hops=2):
34
  super().__init__()
35
  self.cache = "cache.json"
36
- self.llm = dspy.AzureOpenAI(api_base=api_base,
37
- api_version=api_version,
38
  model="GPT-4o")
39
 
40
  self.generate_query = [dspy.ChainOfThought(GenerateSearchQuery) for _ in range(max_hops)]
 
6
  from langchain_community.document_loaders import PyPDFLoader
7
  from langchain_text_splitters import RecursiveCharacterTextSplitter
8
 
9
+
 
10
 
11
 
12
 
 
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)]