From cd411c2eeae1180b3880da1d6fe8c6c341ed60f3 Mon Sep 17 00:00:00 2001
From: binary-husky <505030475@qq.com>
Date: Thu, 25 May 2023 15:12:54 +0800
Subject: [PATCH] newbing-free deps

---
 config.py                    | 2 +-
 request_llm/edge_gpt_free.py | 2 --
 requirements.txt             | 3 ++-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/config.py b/config.py
index 73950ba..14b089e 100644
--- a/config.py
+++ b/config.py
@@ -46,7 +46,7 @@ MAX_RETRY = 2
 
 # 模型选择是 (注意: LLM_MODEL是默认选中的模型, 同时它必须被包含在AVAIL_LLM_MODELS切换列表中 )
 LLM_MODEL = "gpt-3.5-turbo" # 可选 ↓↓↓
-AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm", "moss", "newbing", "stack-claude"]
+AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm", "moss", "newbing", "newbing-free", "stack-claude"]
 # P.S. 其他可用的模型还包括 ["newbing-free", "jittorllms_rwkv", "jittorllms_pangualpha", "jittorllms_llama"]
 
 # 本地LLM模型如ChatGLM的执行方式 CPU/GPU
diff --git a/request_llm/edge_gpt_free.py b/request_llm/edge_gpt_free.py
index 0026db2..7e893d4 100644
--- a/request_llm/edge_gpt_free.py
+++ b/request_llm/edge_gpt_free.py
@@ -28,8 +28,6 @@ from typing import Union
 import aiohttp
 import certifi
 import httpx
-from BingImageCreator import ImageGen
-from BingImageCreator import ImageGenAsync
 from prompt_toolkit import PromptSession
 from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
 from prompt_toolkit.completion import WordCompleter
diff --git a/requirements.txt b/requirements.txt
index 10f9e07..2aedd6e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,6 +4,7 @@ requests[socks]
 transformers
 python-markdown-math
 beautifulsoup4
+prompt_toolkit
 latex2mathml
 python-docx
 mdtex2html
@@ -14,4 +15,4 @@ pymupdf
 openai
 numpy
 arxiv
-
+rich