alex/nova-api
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-api.git synced 2024-11-25 19:33:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed non-streamed text completions

Browse source
This commit is contained in:
nsde 2023-08-22 20:03:55 +02:00
parent 998139d4d8
commit a7b2ce7aa5
4 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
api/db/logs.py
View file

@ -53,7 +53,7 @@ async def log_api_request(user: dict, incoming_request, target_url: str):
model = payload.get('model')
ip_address = await network.get_ip(incoming_request)
useragent = await replacer(incoming_request.headers.get('User-Agent'), UA_SIMPLIFY)
useragent = await replacer(incoming_request.headers.get('User-Agent', ''), UA_SIMPLIFY)
new_log_item = {
'timestamp': time.time(),

2
api/main.py
View file

@ -39,4 +39,4 @@ async def root():
'core_api_docs_for_nova_developers': '/docs'
}
app.add_route('/{path:path}', transfer.handle, ['GET', 'POST', 'PUT', 'DELETE', 'PATCH'])
app.add_route('/v1/{path:path}', transfer.handle, ['GET', 'POST', 'PUT', 'DELETE', 'PATCH'])

2
api/streaming.py
View file

@ -116,7 +116,7 @@ async def stream(
cookies=target_request.get('cookies'),
ssl=False,
timeout=aiohttp.ClientTimeout(
connect=3.0,
connect=60,
total=float(os.getenv('TRANSFER_TIMEOUT', '120'))
),
) as response:

4
api/transfer.py
View file

@ -14,7 +14,7 @@ from helpers import tokens, errors
load_dotenv()
models_list = json.load(open('models.json'))
models_list = json.load(open('models.json', encoding='utf8'))
with open('config/config.yml', encoding='utf8') as f:
config = yaml.safe_load(f)
@ -26,7 +26,7 @@ async def handle(incoming_request):
Checks method, token amount, auth and cost along with if request is NSFW.
"""
users = UserManager()
path = incoming_request.url.path.replace('v1/v1/', 'v1/')
path = incoming_request.url.path
if '/models' in path:
return fastapi.responses.JSONResponse(content=models_list)