Skip to content

Commit f85e79c

Browse files
committed
Solved: The provider cannot correctly change
1 parent 13e0750 commit f85e79c

File tree

3 files changed

+6
-10
lines changed

3 files changed

+6
-10
lines changed

app.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def get_response(query: str) -> str:
139139
'Something wrong in docGPT...\n'
140140
'1. If you are using gpt4free model, '
141141
'try to select the different provider. '
142-
'(Or wait a moment, maybe there have something wrong in gpt4free)\n'
142+
'(DeepAI, ChatgptAi are more stable)\n'
143143
'2. If you are using openai model, '
144144
'check your usage for openai api key.'
145145
)

docGPT/__init__.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -56,12 +56,7 @@ def create_doc_gpt(
5656
return docGPT
5757
else:
5858
# Use gpt4free llm model without agent
59-
llm_model = GPT4Free(
60-
provider=GPT4Free().PROVIDER_MAPPING[
61-
g4f_provider
62-
]
63-
)
64-
print(GPT4Free().PROVIDER_MAPPING[g4f_provider])
59+
llm_model = GPT4Free(provider=g4f_provider)
6560
docGPT.llm = llm_model
6661
docGPT.create_qa_chain(chain_type='refine')
6762
return docGPT

docGPT/docGPT.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,7 @@ class GPT4Free(LLM):
220220
'g4f.Provider.You': g4f.Provider.You,
221221
'g4f.Provider.Yqcloud': g4f.Provider.Yqcloud,
222222
}
223-
provider = PROVIDER_MAPPING['g4f.Provider.ChatgptAi']
223+
provider: str = 'g4f.Provider.ChatgptAi'
224224

225225
@property
226226
def _llm_type(self) -> str:
@@ -233,11 +233,12 @@ def _call(
233233
run_manager: Optional[CallbackManagerForLLMRun] = None,
234234
) -> str:
235235
try:
236-
print('promopt: ', prompt)
236+
print(f'Promopt: {prompt}')
237+
print(f'Provider: {self.PROVIDER_MAPPING[self.provider]}')
237238
return g4f.ChatCompletion.create(
238239
model="gpt-3.5-turbo",
239240
messages=[{"role": "user", "content": prompt}],
240-
provider=self.provider
241+
provider=self.PROVIDER_MAPPING[self.provider]
241242
)
242243
except Exception as e:
243244
print(f'{__file__}: {e}')

0 commit comments

Comments
 (0)