deepanything 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

deepanything/Server/Server.py

@@ -1,3 +1,4 @@
+from chunk import Chunk
 from dataclasses import dataclass
 import time
 import uvicorn
@@ -5,9 +6,10 @@ from typing import Dict, List, Optional, Any
 import json
 
 from openai.types.model import Model as OpenaiModel
-from fastapi import FastAPI,Depends, HTTPException, status,Header
-from fastapi.responses import StreamingResponse
+from fastapi import FastAPI,Depends, HTTPException, status,Header,Request
+from fastapi.responses import StreamingResponse,Response
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+from uvicorn.config import LOGGING_CONFIG
 
 from deepanything.DeepAnythingClient import chat_completion_stream_async, chat_completion_async
 from deepanything.ResponseClient import AsyncOpenaiResponseClient,AsyncResponseClient
@@ -35,6 +37,7 @@ class DeepAnythingServer:
     model_owner : str = "deepanything"
     api_keys : List[str] = []
     security = HTTPBearer()
+    log_config : Dict[str,Any] = LOGGING_CONFIG
 
     def __init__(self, host:str = None, port:int = None, config : Any or str = None):
         if config is not None:
@@ -52,7 +55,7 @@ class DeepAnythingServer:
         self.app.add_api_route("/v1/chat/completions",self.chat_completions,methods=["POST"])
 
     def run(self):
-        uvicorn.run(self.app,host=self.host,port=self.port,log_level="trace")
+        uvicorn.run(self.app,host=self.host,port=self.port,log_config=self.log_config)
 
     @staticmethod
     def _extract_args(query : Types.ChatCompletionQuery) -> dict:
@@ -66,36 +69,18 @@ class DeepAnythingServer:
         self.port = config_object.get("port",8000)
         self.model_owner = config_object.get("model_owner","deepanything")
 
-        reason_clients:List[Dict] = config_object.get("reason_clients",[])
+        self._load_reason_clients(config_object)
+        self._load_response_clients(config_object)
+        self._load_models(config_object)
 
-        for client in reason_clients:
-            name = client["name"]
-            base_url = client["base_url"]
-            api_key = client.get("api_key","")
-            extract_args = client.get("extract_args",{})
-
-            if client["type"] == 'deepseek':
-                self.reason_clients[name] = AsyncDeepseekReasonClient(base_url, api_key, **extract_args)
-            elif client["type"] == 'openai':
-                self.reason_clients[name] = AsyncOpenaiReasonClient(base_url, api_key, **extract_args)
-            else:
-                raise Exception("unknown reason client type")
-
-        response_clients : List[Dict] = config_object.get("response_clients",[])
-
-        for client in response_clients:
-            name = client["name"]
-            base_url = client["base_url"]
-            api_key = client.get("api_key","")
-            extract_args = client.get("extract_args",{})
-
-            if client["type"] == 'openai':
-                self.response_clients[name] = AsyncOpenaiResponseClient(base_url,api_key,**extract_args)
-            else:
-                raise Exception("unknown response client type")
+        self.api_keys = config_object.get("api_keys",[])
+        self.log_config = config_object.get("log",LOGGING_CONFIG)
+        if self.log_config == {}:
+            self.log_config = LOGGING_CONFIG
 
-        models : List[Dict] = config_object.get("models",[])
 
+    def _load_models(self, config_object):
+        models: List[Dict] = config_object.get("models", [])
         for _model in models:
             name = _model["name"]
             reason_client = _model["reason_client"]
@@ -103,20 +88,51 @@ class DeepAnythingServer:
             response_client = _model["response_client"]
             response_model = _model["response_model"]
             created = _model.get("created", int(time.time()))
-            reason_prompt = _model.get("reason_prompt","<Think>{}</Think>")
+            reason_prompt = _model.get("reason_prompt", "<Think>{}</Think>")
+
+            if reason_client not in self.reason_clients:
+                raise ValueError(f"Reason client '{reason_model}' for '{name}' not found")
+
+            if response_client not in self.response_clients:
+                raise ValueError(f"Response client '{response_model}' for '{name}' not found")
 
             self.models[name] = ModelInfo(
-                name = name,
-                reason_client = reason_client,
-                reason_model = reason_model,
-                response_client = response_client,
-                response_model = response_model,
-                created = created,
-                reason_prompt = reason_prompt
+                name=name,
+                reason_client=reason_client,
+                reason_model=reason_model,
+                response_client=response_client,
+                response_model=response_model,
+                created=created,
+                reason_prompt=reason_prompt
             )
 
-        self.api_keys = config_object.get("api_keys",[])
+    def _load_response_clients(self, config_object):
+        response_clients: List[Dict] = config_object.get("response_clients", [])
+        for client in response_clients:
+            name = client["name"]
+            base_url = client["base_url"]
+            api_key = client.get("api_key", "")
+            extract_args = client.get("extract_args", {})
+
+            if client["type"] == 'openai':
+                self.response_clients[name] = AsyncOpenaiResponseClient(base_url, api_key, **extract_args)
+            else:
+                raise ValueError(f"Unsupported response client type '{client['type']}'")
 
+    def _load_reason_clients(self, config_object):
+        reason_clients: List[Dict] = config_object.get("reason_clients", [])
+        for client in reason_clients:
+            name = client["name"]
+            base_url = client["base_url"]
+            api_key = client.get("api_key", "")
+            extract_args = client.get("extract_args", {})
+
+            if client["type"] == 'deepseek':
+                self.reason_clients[name] = AsyncDeepseekReasonClient(base_url, api_key, **extract_args)
+            elif client["type"] == 'openai':
+                self.reason_clients[name] = AsyncOpenaiReasonClient(base_url, api_key, **extract_args)
+            else:
+                raise Exception("unknown reason client type")
 
     def add_reason_client(self,name:str,client:AsyncReasonClient):
         self.reason_clients[name] = client
@@ -126,7 +142,8 @@ class DeepAnythingServer:
 
     def add_model(self,name:str,model:ModelInfo):
         self.models[name] = model
-    def verify_authorization(self, authorization:Optional[str]):
+
+    def _verify_authorization(self, authorization:Optional[str]):
         if not self.api_keys:
             return
 
@@ -152,13 +169,29 @@ class DeepAnythingServer:
                 headers={"WWW-Authenticate": "Bearer"},
             )
 
-    async def chat_completions(self, query : Types.ChatCompletionQuery, authorization:Optional[str] = Header(None)):
-        self.verify_authorization(authorization)
+    async def chat_completions(
+            self,
+            request: Request,  # 新增加Request参数
+            query: Types.ChatCompletionQuery,
+            authorization: Optional[str] = Header(None)
+    ):
+        self._verify_authorization(authorization)
+
+        if query.model not in self.models:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail="Model not found",
+            )
+
         model = self.models[query.model]
-        async def sse(it: AsyncStream):
+
+        async def _sse_warp(it: AsyncStream, req: Request):
             async for chunk in it:
-                yield f"data: {chunk.model_dump_json(indent=None)}\n\n"
-            yield "data: [DONE]"
+                if await req.is_disconnected():
+                    await it.close()
+                    break
+                yield f"data: {chunk.model_dump_json(indent=None)}\n\n".encode("utf-8")
+            yield "data: [DONE]".encode('utf-8')
 
         args = DeepAnythingServer._extract_args(query)
 
@@ -168,7 +201,7 @@ class DeepAnythingServer:
             args.pop("max_tokens")
 
         if query.stream:
-            res = sse(
+            res = _sse_warp(
                 await chat_completion_stream_async(
                     messages=query.messages,
                     reason_client=self.reason_clients[model.reason_client],
@@ -180,26 +213,30 @@ class DeepAnythingServer:
                     response_args=args,
                     reason_args=args,
                     max_tokens=max_tokens
-                )
+                ),
+                request
             )
             return StreamingResponse(
                 res,
-                media_type="text/event-stream"
+                media_type="text/event-stream",
             )
         else:
             res = await chat_completion_async(
-                    messages=query.messages,
-                    reason_client=self.reason_clients[model.reason_client],
-                    reason_model=model.reason_model,
-                    response_client=self.response_clients[model.response_client],
-                    response_model=model.response_model,
-                    show_model=model.name,
-                    reason_prompt=model.reason_prompt,
-                    response_args=args,
-                    reason_args=args,
-                    max_tokens=max_tokens
+                messages=query.messages,
+                reason_client=self.reason_clients[model.reason_client],
+                reason_model=model.reason_model,
+                response_client=self.response_clients[model.response_client],
+                response_model=model.response_model,
+                show_model=model.name,
+                reason_prompt=model.reason_prompt,
+                response_args=args,
+                reason_args=args,
+                max_tokens=max_tokens
+            )
+            return Response(
+                content=res.model_dump_json(indent=None),
+                media_type="application/json"
             )
-            return res.model_dump_json()
 
     def get_models(self) -> Types.ModelsListResponse:
         return Types.ModelsListResponse(

deepanything/__main__.py

@@ -1,11 +1,13 @@
 from deepanything.Server.Server import DeepAnythingServer
 import argparse
 import json
+from .metadatas import VERSION
 def main():
-    parser = argparse.ArgumentParser(prog="deepanything",description="Run a DeepAnything Server.")
+    parser = argparse.ArgumentParser(prog=f"deepanything {VERSION}",description="Run a DeepAnything Server.")
     parser.add_argument('--host', type=str, required=False, help='Specific the host to listen.If specified,the host will be overwritten by this.')
     parser.add_argument('--port', type=int, required=False, help='Specific the port to listen.If specified,the port will be overwritten by this.')
     parser.add_argument('--config', type=str, required=True, help='Specific the confi path.')
+    parser.add_argument('--version', action='version', version=f'%(prog)s {VERSION}')
 
     args = parser.parse_args()
 

deepanything/metadatas.py

@@ -0,0 +1 @@
+VERSION = "v0.1.6"

{deepanything-0.1.4.dist-info → deepanything-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: deepanything
-Version: 0.1.4
+Version: 0.1.6
 Summary: DeepAnything is a project that provides DeepSeek R1's deep thinking capabilities for various large language models (LLMs).
 Author: Junity
 Author-email: 1727636624@qq.com
@@ -159,6 +159,7 @@ More example can be find in [examples](examples).
 Below is an example of a configuration file:
 
 ```json
+// Using R1 with Qwen-Max-Latest
 {
   "host" : "0.0.0.0",
   "port" : 8080,
@@ -187,9 +188,41 @@ Below is an example of a configuration file:
       "response_model" :  "qwen-max-latest"
     }
   ],
-  "api_keys": [
+  "api_keys" : [
     "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
-  ]
+  ],
+  "log": {
+    "version": 1,
+    "disable_existing_loggers": false,
+    "formatters": {
+        "default": {
+            "()": "uvicorn.logging.DefaultFormatter",
+            "fmt": "%(levelprefix)s %(message)s",
+            "use_colors": null
+        },
+        "access": {
+            "()": "uvicorn.logging.AccessFormatter",
+            "fmt": "%(levelprefix)s %(client_addr)s - \"%(request_line)s\" %(status_code)s"
+        }
+    },
+    "handlers": {
+        "default": {
+            "formatter": "default",
+            "class": "logging.StreamHandler",
+            "stream": "ext://sys.stderr"
+        },
+        "access": {
+            "formatter": "access",
+            "class": "logging.StreamHandler",
+            "stream": "ext://sys.stdout"
+        }
+    },
+    "loggers": {
+        "uvicorn": {"handlers": ["default"], "level": "INFO", "propagate": false},
+        "uvicorn.error": {"level": "INFO"},
+        "uvicorn.access": {"handlers": ["access"], "level": "INFO", "propagate": false}
+    }
+  }
 }
 ```
 
@@ -198,6 +231,26 @@ Below is an example of a configuration file:
 - reason_clients: Configuration for thinking models, currently supports deepseek and openai types. When the type is openai, deepanything directly uses the model's output as the thinking content, and it is recommended to use qwq-32b in this case.
 - response_clients: Configuration for response models, currently only supports the openai type.
 - api_keys: API keys for user authentication. When left blank or an empty list, the server does not use API keys for authentication.
+- log: Log configuration. If this item is not filled in, the default logging configuration of uvicorn will be used. For details, please refer to [uvicorn logging configuration](https://www.uvicorn.org/settings/#logging).
+
+## Deploying with Docker
+### 1. Pull the Image
+```bash
+docker pull junity233/deepanything:latest
+```
+
+### 2. Create config.json
+First, create a folder in your desired directory, and then create a file named `config.json` inside it. This folder will be mounted into the container:
+```bash
+mkdir deepanything-data               # You can replace this with another name
+vim deepanything-data/config.json     # Edit the configuration file, you can refer to examples/config.json
+```
+
+### 3. Run the Container
+```bash
+# Remember to modify the port mapping
+docker run -v ./deepanything-data:/data -p 8080:8080 junity233/deepanything:latest
+```
 
 ## License
 

{deepanything-0.1.4.dist-info → deepanything-0.1.6.dist-info}/RECORD

@@ -4,13 +4,14 @@ deepanything/ResponseClient.py,sha256=NbXjlU_0qTKBNjZy8B9J9emuABQYvx3NZsWuja9OnM
 deepanything/Stream.py,sha256=8ESR8ttjyPZ-uXPDENsVWUzaL34_GT2OZBJ0PWu7vsA,1578
 deepanything/Utility.py,sha256=TFH-4NS-gQpzzEb1Aba7WpDEqBqNRF1STWFzfRqQLcg,6645
 deepanything/__init__.py,sha256=_2RolcKcpxmW0dmtiQpXlvgxe5dvqx90Yg_Q_oVLVZQ,175
-deepanything/__main__.py,sha256=A6SQ1hkW2LAiHoRxQJdCLrSGUn3NP-eQY8O2oTv9nYA,936
-deepanything/Server/Server.py,sha256=90MMxWXppLuUy8o9lEyJC_yXJeSSanBDBOjCcsG0yr8,8359
+deepanything/__main__.py,sha256=N40zRebnTUmVd9a0wcK_I0au7Lvl4axxKX6zJGDdXq0,1066
+deepanything/metadatas.py,sha256=WspkzXRul9GiZA9N8kBo-wnZEexn6N3Y_ykPEaT8dL8,18
+deepanything/Server/Server.py,sha256=HIM-MHuRG98jv8MTeGK9So-xd8OiTA9R74uvKbWrXyk,9847
 deepanything/Server/Types.py,sha256=b7aMaRBgODEKdyYe0FeraUfrygJuye3b5lfQTOWASXA,650
 deepanything/Server/__init__.py,sha256=eIpn6NbNvEg4ST8CuuIuzPT3m_fTlmPC3sikPoPFsYo,92
-deepanything-0.1.4.dist-info/LICENSE,sha256=JWYd2E-mcNcSYjT5nk4ayM5kkkDq6ZlOxVcYsyqCIwU,1059
-deepanything-0.1.4.dist-info/METADATA,sha256=7MKrxbIcsqBLCpb6CBppWi7ftEMHnNmC829IBAMKCUg,5893
-deepanything-0.1.4.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-deepanything-0.1.4.dist-info/entry_points.txt,sha256=UT4gNGx6dJsKBjZIl3VkMekh385O5WMbMidAAla6UB4,60
-deepanything-0.1.4.dist-info/top_level.txt,sha256=wGeRb__4jEJTclCUl0cxhgubD_Bq-QT38VIH6C4KpzY,13
-deepanything-0.1.4.dist-info/RECORD,,
+deepanything-0.1.6.dist-info/LICENSE,sha256=JWYd2E-mcNcSYjT5nk4ayM5kkkDq6ZlOxVcYsyqCIwU,1059
+deepanything-0.1.6.dist-info/METADATA,sha256=-h3eS_CkiwY27ju-QxH2_40IJMBszckjbjmhBTqW0Ig,7851
+deepanything-0.1.6.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+deepanything-0.1.6.dist-info/entry_points.txt,sha256=UT4gNGx6dJsKBjZIl3VkMekh385O5WMbMidAAla6UB4,60
+deepanything-0.1.6.dist-info/top_level.txt,sha256=wGeRb__4jEJTclCUl0cxhgubD_Bq-QT38VIH6C4KpzY,13
+deepanything-0.1.6.dist-info/RECORD,,