Create Ava.py

g4f/Provider/Providers/Ava.py

@@ -0,0 +1,61 @@
+import requests
+import json
+import os
+from ...typing import sha256, Dict, get_type_hints
+
+url = "https://ava-alpha-api.codelink.io/api/chat"
+model = ["gpt-4"]
+supports_stream = True
+needs_auth = False
+
+class Model:
+    def __init__(self):
+        self.url = "https://ava-alpha-api.codelink.io/api/chat"
+        self.headers = {
+            "content-type": "application/json"
+        }
+        self.payload = {
+            "model": "gpt-4",
+            "temperature": 0.6,
+            "stream": True
+        }
+        self.accumulated_content = ""
+
+    def _process_line(self, line):
+        line_text = line.decode("utf-8").strip()
+        if line_text.startswith("data:"):
+            data = line_text[len("data:"):]
+            try:
+                data_json = json.loads(data)
+                if "choices" in data_json:
+                    choices = data_json["choices"]
+                    for choice in choices:
+                        if "finish_reason" in choice and choice["finish_reason"] == "stop":
+                            break
+                        if "delta" in choice and "content" in choice["delta"]:
+                            content = choice["delta"]["content"]
+                            self.accumulated_content += content
+            except json.JSONDecodeError as e:
+                return
+
+    def ChatCompletion(self, messages):
+        self.payload["messages"] = messages
+
+        with requests.post(self.url, headers=self.headers, data=json.dumps(self.payload), stream=True) as response:
+            for line in response.iter_lines():
+                self._process_line(line)
+
+        accumulated_content = self.accumulated_content
+        self.accumulated_content = ""
+
+        return accumulated_content
+    
+def _create_completion(model: str, messages: list, stream: bool, **kwargs):
+    model = Model()
+
+    # Call the chat completion method
+    response = model.ChatCompletion(messages)
+    return response
+
+params = f'g4f.Providers.{os.path.basename(__file__)[:-3]} supports: ' + \
+    '(%s)' % ', '.join([f"{name}: {get_type_hints(_create_completion)[name].__name__}" for name in _create_completion.__code__.co_varnames[:_create_completion.__code__.co_argcount]])