File tree Expand file tree Collapse file tree 3 files changed +6
-10
lines changed Expand file tree Collapse file tree 3 files changed +6
-10
lines changed Original file line number Diff line number Diff line change @@ -139,7 +139,7 @@ def get_response(query: str) -> str:
139
139
'Something wrong in docGPT...\n '
140
140
'1. If you are using gpt4free model, '
141
141
'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 '
143
143
'2. If you are using openai model, '
144
144
'check your usage for openai api key.'
145
145
)
Original file line number Diff line number Diff line change @@ -56,12 +56,7 @@ def create_doc_gpt(
56
56
return docGPT
57
57
else :
58
58
# 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 )
65
60
docGPT .llm = llm_model
66
61
docGPT .create_qa_chain (chain_type = 'refine' )
67
62
return docGPT
Original file line number Diff line number Diff line change @@ -220,7 +220,7 @@ class GPT4Free(LLM):
220
220
'g4f.Provider.You' : g4f .Provider .You ,
221
221
'g4f.Provider.Yqcloud' : g4f .Provider .Yqcloud ,
222
222
}
223
- provider = PROVIDER_MAPPING [ 'g4f.Provider.ChatgptAi' ]
223
+ provider : str = 'g4f.Provider.ChatgptAi'
224
224
225
225
@property
226
226
def _llm_type (self ) -> str :
@@ -233,11 +233,12 @@ def _call(
233
233
run_manager : Optional [CallbackManagerForLLMRun ] = None ,
234
234
) -> str :
235
235
try :
236
- print ('promopt: ' , prompt )
236
+ print (f'Promopt: { prompt } ' )
237
+ print (f'Provider: { self .PROVIDER_MAPPING [self .provider ]} ' )
237
238
return g4f .ChatCompletion .create (
238
239
model = "gpt-3.5-turbo" ,
239
240
messages = [{"role" : "user" , "content" : prompt }],
240
- provider = self .provider
241
+ provider = self .PROVIDER_MAPPING [ self . provider ]
241
242
)
242
243
except Exception as e :
243
244
print (f'{ __file__ } : { e } ' )
You can’t perform that action at this time.
0 commit comments