From b0276f6c9e2925c65a00b7189bad92feb25cefbd Mon Sep 17 00:00:00 2001
From: madonchik123 <68397448+madonchik123@users.noreply.github.com>
Date: Sat, 2 Dec 2023 01:11:52 +0300
Subject: Added new provider PI (Hacky way to use) (#1291)

* Added new provider PI (Hacky way to use)

* Updated models endpoint made it show real info about the model.py

* Added cloudscraper to the requirements

* Fixed some bugs aka made streaming also return role
---
 g4f/models.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'g4f/models.py')

diff --git a/g4f/models.py b/g4f/models.py
index 46eb49a0..e4b20ae5 100644
--- a/g4f/models.py
+++ b/g4f/models.py
@@ -27,6 +27,7 @@ from .Provider   import (
     Bing,
     You,
     H2o,
+    PI,
 )
 
 @dataclass(unsafe_hash=True)
@@ -260,6 +261,11 @@ llama70b_v2_chat = Model(
     base_provider = 'replicate',
     best_provider = Vercel)
 
+pi = Model(
+    name = 'pi',
+    base_provider = 'inflection',
+    best_provider=PI
+)
 
 class ModelUtils:
     convert: dict[str, Model] = {
@@ -315,6 +321,8 @@ class ModelUtils:
         'oasst-sft-1-pythia-12b'           : oasst_sft_1_pythia_12b,
         'oasst-sft-4-pythia-12b-epoch-3.5' : oasst_sft_4_pythia_12b_epoch_35,
         'command-light-nightly'            : command_light_nightly,
+
+        'pi': pi
     }
 
 _all_models = list(ModelUtils.convert.keys())
\ No newline at end of file
-- 
cgit v1.2.3