gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed linting

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-01-21 00:10:52 +01:00
parent 710ca687c2
commit a3311df27c
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
extra/OpenWebuiTool/openwebui_tool.py
View File

@@ -30,11 +30,8 @@ __author_email__ = "parisneoai@gmail.com"
__description__ = "Lightrag integration for OpenWebui" __description__ = "Lightrag integration for OpenWebui"
import os
import requests import requests
import json import json
import csv
from io import StringIO
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from typing import Callable, Any, Literal, Union, List, Tuple from typing import Callable, Any, Literal, Union, List, Tuple
@@ -238,7 +235,7 @@ class Tools:
remaining = remaining[index + 3 :] remaining = remaining[index + 3 :]
first_index += index + 3 first_index += index + 3
bloc_index += 1 bloc_index += 1
except Exception as ex: except Exception:
if bloc_index % 2 == 1: if bloc_index % 2 == 1:
index = len(remaining) index = len(remaining)
indices.append(index) indices.append(index)
@@ -293,11 +290,11 @@ class Tools:
if len(sub_text) > 0: if len(sub_text) > 0:
try: try:
find_space = sub_text.index(" ") find_space = sub_text.index(" ")
except: except Exception:
find_space = int(1e10) find_space = int(1e10)
try: try:
find_return = sub_text.index("\n") find_return = sub_text.index("\n")
except: except Exception:
find_return = int(1e10) find_return = int(1e10)
next_index = min(find_return, find_space) next_index = min(find_return, find_space)
if "{" in sub_text[:next_index]: if "{" in sub_text[:next_index]: