apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-05 05:42:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

chore: ran linting

Browse source
This commit is contained in:
Anish Sarkar 2026-02-04 03:04:25 +05:30
parent 30c6f42102
commit 580b75c3c4
3 changed files with 40 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

41
surfsense_backend/app/connectors/composio_google_drive_connector.py
View file

@ -399,7 +399,9 @@ async def _extract_text_with_etl(
from logging import ERROR, getLogger
etl_service = config.ETL_SERVICE
logger.debug(f"[_extract_text_with_etl] START - file_path={file_path}, file_name={file_name}, etl_service={etl_service}")
logger.debug(
f"[_extract_text_with_etl] START - file_path={file_path}, file_name={file_name}, etl_service={etl_service}"
)
try:
if etl_service == "UNSTRUCTURED":
@ -419,10 +421,14 @@ async def _extract_text_with_etl(
)
docs = await loader.aload()
logger.debug(f"[_extract_text_with_etl] UNSTRUCTURED loaded {len(docs) if docs else 0} docs")
logger.debug(
f"[_extract_text_with_etl] UNSTRUCTURED loaded {len(docs) if docs else 0} docs"
)
if docs:
result = await convert_document_to_markdown(docs)
logger.debug(f"[_extract_text_with_etl] UNSTRUCTURED result: {len(result) if result else 0} chars")
logger.debug(
f"[_extract_text_with_etl] UNSTRUCTURED result: {len(result) if result else 0} chars"
)
return result
logger.debug("[_extract_text_with_etl] UNSTRUCTURED returned no docs")
return None
@ -447,12 +453,18 @@ async def _extract_text_with_etl(
markdown_documents = await result.aget_markdown_documents(
split_by_page=False
)
logger.debug(f"[_extract_text_with_etl] LLAMACLOUD got {len(markdown_documents) if markdown_documents else 0} markdown docs")
logger.debug(
f"[_extract_text_with_etl] LLAMACLOUD got {len(markdown_documents) if markdown_documents else 0} markdown docs"
)
if markdown_documents:
text = markdown_documents[0].text
logger.debug(f"[_extract_text_with_etl] LLAMACLOUD result: {len(text) if text else 0} chars")
logger.debug(
f"[_extract_text_with_etl] LLAMACLOUD result: {len(text) if text else 0} chars"
)
return text
logger.debug("[_extract_text_with_etl] LLAMACLOUD returned no markdown docs")
logger.debug(
"[_extract_text_with_etl] LLAMACLOUD returned no markdown docs"
)
return None
elif etl_service == "DOCLING":
@ -480,20 +492,29 @@ async def _extract_text_with_etl(
result = await docling_service.process_document(
file_path, file_name
)
logger.debug(f"[_extract_text_with_etl] DOCLING result keys: {list(result.keys()) if result else 'None'}")
logger.debug(
f"[_extract_text_with_etl] DOCLING result keys: {list(result.keys()) if result else 'None'}"
)
finally:
pdfminer_logger.setLevel(original_level)
content = result.get("content")
logger.debug(f"[_extract_text_with_etl] DOCLING content: {len(content) if content else 0} chars")
logger.debug(
f"[_extract_text_with_etl] DOCLING content: {len(content) if content else 0} chars"
)
return content
else:
logger.warning(f"[_extract_text_with_etl] Unknown ETL service: {etl_service}")
logger.warning(
f"[_extract_text_with_etl] Unknown ETL service: {etl_service}"
)
return None
except Exception as e:
logger.error(f"[_extract_text_with_etl] ETL extraction EXCEPTION for {file_name}: {e!s}")
logger.error(
f"[_extract_text_with_etl] ETL extraction EXCEPTION for {file_name}: {e!s}"
)
import traceback
logger.error(f"[_extract_text_with_etl] Traceback: {traceback.format_exc()}")
return None

15
surfsense_web/app/dashboard/[search_space_id]/new-chat/[[...chat_id]]/page.tsx
View file

@ -390,7 +390,7 @@ export default function NewChatPage() {
}
setIsRunning(false);
}, []);
// Handle new message from user
const onNew = useCallback(
async (message: AppendMessage) => {
@ -1357,14 +1357,11 @@ export default function NewChatPage() {
);
// Handle reloading/refreshing the last AI response
const onReload = useCallback(
async () => {
// parentId is the ID of the message to reload from (the user message)
// We call regenerate without a query to use the same query
await handleRegenerate(null);
},
[handleRegenerate]
);
const onReload = useCallback(async () => {
// parentId is the ID of the message to reload from (the user message)
// We call regenerate without a query to use the same query
await handleRegenerate(null);
}, [handleRegenerate]);
// Create external store runtime with attachment support
const runtime = useExternalStoreRuntime({

6
surfsense_web/components/new-chat/document-mention-picker.tsx
View file

@ -417,8 +417,8 @@ export const DocumentMentionPicker = forwardRef<
<Skeleton className="h-[16px] w-24" />
</div>
{["a", "b", "c", "d", "e"].map((id, i) => (
<div
key={id}
<div
key={id}
className={cn(
"w-full flex items-center gap-2 px-3 py-2 text-left rounded-md",
i >= 3 && "hidden sm:flex"
@ -428,7 +428,7 @@ export const DocumentMentionPicker = forwardRef<
<Skeleton className="h-4 w-4" />
</span>
<span className="flex-1 text-sm">
<Skeleton className="h-[20px]" style={{ width: `${60 + (i * 7) % 30}%` }} />
<Skeleton className="h-[20px]" style={{ width: `${60 + ((i * 7) % 30)}%` }} />
</span>
</div>
))}