bugfix: doc_parser with windows path

ye7love7 · Feb 27, 2024 · 6b73b72 · 6b73b72
1 parent aa0cc17
commit 6b73b72
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/qwen_agent/tools/doc_parser.py b/qwen_agent/tools/doc_parser.py
@@ -77,7 +77,8 @@ def process_file(url: str, db: Storage = None):
     if url.split('.')[-1].lower() in ['pdf', 'docx', 'pptx']:
         date1 = datetime.datetime.now()
 
-        if url.startswith('https://') or url.startswith('http://'):
+        if url.startswith('https://') or url.startswith('http://') or re.match(
+                r'^[A-Za-z]:\\', url) or re.match(r'^[A-Za-z]:/', url):
             pdf_path = url
         else:
             parsed_url = urlparse(url)