Update models_for_langchain/model.py

models_for_langchain/model.py

@@ -77,11 +77,13 @@ class CustomLLM(LLM):
         #     raise ValueError("stop kwargs are not permitted.")
         provider_llm = getattr(Provider, self.provider_name)
         provider_llm.working = True
+
         bot_msg = g4f.ChatCompletion.create(model=self.model_name, 
                                             provider=provider_dict[self.provider_name], 
                                             messages=[{"role": "user", 
                                                         "content": prompt}],
-                                            stream=False,auth=provider_auth_settings['provider'] if provider_llm.needs_auth else None)
+                                            stream=provider_llm.supports_stream,
+                                            auth=provider_auth_settings['provider'] if provider_llm.needs_auth else None)
         return bot_msg
 
     @property