Spaces:
Sleeping
Sleeping
Update dspy_qa.py
Browse files- 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 |
-
|
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=
|
37 |
-
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)]
|