chore: format
This commit is contained in:
@@ -178,7 +178,11 @@ class DoclingLoader:
|
|||||||
|
|
||||||
params["force_ocr"] = self.params.get("force_ocr")
|
params["force_ocr"] = self.params.get("force_ocr")
|
||||||
|
|
||||||
if self.params.get("do_ocr") and self.params.get("ocr_engine") and self.params.get("ocr_lang"):
|
if (
|
||||||
|
self.params.get("do_ocr")
|
||||||
|
and self.params.get("ocr_engine")
|
||||||
|
and self.params.get("ocr_lang")
|
||||||
|
):
|
||||||
params["ocr_engine"] = self.params.get("ocr_engine")
|
params["ocr_engine"] = self.params.get("ocr_engine")
|
||||||
params["ocr_lang"] = [
|
params["ocr_lang"] = [
|
||||||
lang.strip()
|
lang.strip()
|
||||||
@@ -195,7 +199,6 @@ class DoclingLoader:
|
|||||||
if self.params.get("pipeline"):
|
if self.params.get("pipeline"):
|
||||||
params["pipeline"] = self.params.get("pipeline")
|
params["pipeline"] = self.params.get("pipeline")
|
||||||
|
|
||||||
|
|
||||||
endpoint = f"{self.url}/v1/convert/file"
|
endpoint = f"{self.url}/v1/convert/file"
|
||||||
r = requests.post(endpoint, files=files, data=params)
|
r = requests.post(endpoint, files=files, data=params)
|
||||||
|
|
||||||
|
|||||||
@@ -337,7 +337,10 @@ async def speech(request: Request, user=Depends(get_verified_user)):
|
|||||||
timeout=timeout, trust_env=True
|
timeout=timeout, trust_env=True
|
||||||
) as session:
|
) as session:
|
||||||
r = await session.post(
|
r = await session.post(
|
||||||
url=urljoin(request.app.state.config.TTS_OPENAI_API_BASE_URL, "/audio/speech"),
|
url=urljoin(
|
||||||
|
request.app.state.config.TTS_OPENAI_API_BASE_URL,
|
||||||
|
"/audio/speech",
|
||||||
|
),
|
||||||
json=payload,
|
json=payload,
|
||||||
headers={
|
headers={
|
||||||
"Content-Type": "application/json",
|
"Content-Type": "application/json",
|
||||||
@@ -465,7 +468,10 @@ async def speech(request: Request, user=Depends(get_verified_user)):
|
|||||||
timeout=timeout, trust_env=True
|
timeout=timeout, trust_env=True
|
||||||
) as session:
|
) as session:
|
||||||
async with session.post(
|
async with session.post(
|
||||||
urljoin(base_url or f"https://{region}.tts.speech.microsoft.com", "/cognitiveservices/v1"),
|
urljoin(
|
||||||
|
base_url or f"https://{region}.tts.speech.microsoft.com",
|
||||||
|
"/cognitiveservices/v1",
|
||||||
|
),
|
||||||
headers={
|
headers={
|
||||||
"Ocp-Apim-Subscription-Key": request.app.state.config.TTS_API_KEY,
|
"Ocp-Apim-Subscription-Key": request.app.state.config.TTS_API_KEY,
|
||||||
"Content-Type": "application/ssml+xml",
|
"Content-Type": "application/ssml+xml",
|
||||||
|
|||||||
@@ -340,7 +340,10 @@ def merge_ollama_models_lists(model_lists):
|
|||||||
return list(merged_models.values())
|
return list(merged_models.values())
|
||||||
|
|
||||||
|
|
||||||
@cached(ttl=MODELS_CACHE_TTL, key=lambda _, user: f"ollama_all_models_{user.id}" if user else "ollama_all_models")
|
@cached(
|
||||||
|
ttl=MODELS_CACHE_TTL,
|
||||||
|
key=lambda _, user: f"ollama_all_models_{user.id}" if user else "ollama_all_models",
|
||||||
|
)
|
||||||
async def get_all_models(request: Request, user: UserModel = None):
|
async def get_all_models(request: Request, user: UserModel = None):
|
||||||
log.info("get_all_models()")
|
log.info("get_all_models()")
|
||||||
if request.app.state.config.ENABLE_OLLAMA_API:
|
if request.app.state.config.ENABLE_OLLAMA_API:
|
||||||
|
|||||||
Reference in New Issue
Block a user