Implemented the retry mechanism for the G4FLLM.py script (5 times).

.gitignore

@@ -159,3 +159,4 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 .idea/
 .vscode/
+test.py

langchain_g4f/G4FLLM.py

@@ -8,7 +8,7 @@ from langchain.callbacks.manager import CallbackManagerForLLMRun, AsyncCallbackM
 from langchain.llms.base import LLM
 from langchain.llms.utils import enforce_stop_tokens
 
-
+MAX_TRIES = 5
 class G4FLLM(LLM):
     model: Union[Model, str]
     provider: Optional[type[BaseProvider]] = None
@@ -33,6 +33,8 @@ class G4FLLM(LLM):
         if self.auth is not None:
             create_kwargs["auth"] = self.auth
 
+        for i in range(MAX_TRIES):
+            try:
                 text = ChatCompletion.create(
                     messages=[{"role": "user", "content": prompt}],
                     **create_kwargs,
@@ -42,7 +44,14 @@ class G4FLLM(LLM):
                 text = text if type(text) is str else "".join(text)
                 if stop is not None:
                     text = enforce_stop_tokens(text, stop)
+                if text:
                     return text
+                print(f"Empty response, trying {i+1} of {MAX_TRIES}")
+            except Exception as e:
+                print(f"Error in G4FLLM._call: {e}, trying {i+1} of {MAX_TRIES}")
+        return ""
+    
+
     async def _acall(self, prompt: str, stop: Optional[List[str]] = None, run_manager: Optional[AsyncCallbackManagerForLLMRun] = None, **kwargs: Any) -> str:
         create_kwargs = {} if self.create_kwargs is None else self.create_kwargs.copy()
         create_kwargs["model"] = self.model