botrun-flow-lang 5.12.263__py3-none-any.whl → 6.2.21__py3-none-any.whl

botrun_flow_lang/tests/test_local_files.py

@@ -1,114 +1,114 @@
-import unittest
-import asyncio
-import os
-from pathlib import Path
-from botrun_flow_lang.langgraph_agents.agents.util.local_files import (
-    generate_tmp_text_file,
-    read_tmp_text_file,
-)
-
-
-class TestLocalFiles(unittest.TestCase):
-    def setUp(self):
-        """Set up test fixtures"""
-        # Test configuration
-        self.test_text_content = (
-            "This is a test text file content.\n\nLine 2\nLine 3\n測試中文內容"
-        )
-        self.user_id = "sebastian.hsu@gmail.com"
-
-    def test_generate_tmp_text_file(self):
-        """Test generate_tmp_text_file function"""
-        # Run the async function
-        result = asyncio.run(
-            generate_tmp_text_file(self.test_text_content, self.user_id)
-        )
-
-        # Print the result for manual verification
-        print("\nTest Result:")
-        print(f"Generated storage path: {result}")
-
-        # Basic validation
-        if result.startswith("Error:"):
-            print(f"Test failed with error: {result}")
-            self.fail(f"Test failed with error: {result}")
-        else:
-            print("Test successful - storage path generated")
-            # Validate storage path format
-            expected_path_prefix = f"tmp/{self.user_id}/"
-            self.assertTrue(
-                result.startswith(expected_path_prefix),
-                f"Expected storage path to start with '{expected_path_prefix}', got: {result}",
-            )
-            
-            # Verify content by reading back the file
-            read_content = asyncio.run(read_tmp_text_file(result))
-            if read_content.startswith("Error:"):
-                self.fail(f"Failed to read back content: {read_content}")
-            else:
-                self.assertEqual(
-                    read_content, 
-                    self.test_text_content, 
-                    "Read content doesn't match original content"
-                )
-                print("Content verification successful - read content matches original")
-
-    def test_generate_tmp_text_file_with_unicode(self):
-        """Test generate_tmp_text_file function with unicode content"""
-        unicode_content = "Unicode test: 你好世界 🌍 ñáéíóú âêîôû àèìòù"
-
-        result = asyncio.run(
-            generate_tmp_text_file(unicode_content, self.user_id)
-        )
-
-        print(f"\nUnicode Test Result: {result}")
-
-        if result.startswith("Error:"):
-            print(f"Unicode test failed: {result}")
-            self.fail(f"Unicode test failed: {result}")
-        else:
-            print("Unicode test successful - storage path generated")
-            
-            # Verify content by reading back the file
-            read_content = asyncio.run(read_tmp_text_file(result))
-            if read_content.startswith("Error:"):
-                self.fail(f"Failed to read back unicode content: {read_content}")
-            else:
-                self.assertEqual(
-                    read_content, 
-                    unicode_content, 
-                    "Unicode read content doesn't match original content"
-                )
-                print("Unicode content verification successful - read content matches original")
-
-    def test_generate_tmp_text_file_with_empty_content(self):
-        """Test generate_tmp_text_file function with empty content"""
-        empty_content = ""
-
-        result = asyncio.run(
-            generate_tmp_text_file(empty_content, self.user_id)
-        )
-
-        print(f"\nEmpty Content Test Result: {result}")
-
-        if result.startswith("Error:"):
-            print(f"Empty content test failed: {result}")
-            self.fail(f"Empty content test failed: {result}")
-        else:
-            print("Empty content test successful - storage path generated")
-            
-            # Verify content by reading back the file
-            read_content = asyncio.run(read_tmp_text_file(result))
-            if read_content.startswith("Error:"):
-                self.fail(f"Failed to read back empty content: {read_content}")
-            else:
-                self.assertEqual(
-                    read_content, 
-                    empty_content, 
-                    "Empty read content doesn't match original content"
-                )
-                print("Empty content verification successful - read content matches original")
-
-
-if __name__ == "__main__":
-    unittest.main()
+import unittest
+import asyncio
+import os
+from pathlib import Path
+from botrun_flow_lang.langgraph_agents.agents.util.local_files import (
+    generate_tmp_text_file,
+    read_tmp_text_file,
+)
+
+
+class TestLocalFiles(unittest.TestCase):
+    def setUp(self):
+        """Set up test fixtures"""
+        # Test configuration
+        self.test_text_content = (
+            "This is a test text file content.\n\nLine 2\nLine 3\n測試中文內容"
+        )
+        self.user_id = "sebastian.hsu@gmail.com"
+
+    def test_generate_tmp_text_file(self):
+        """Test generate_tmp_text_file function"""
+        # Run the async function
+        result = asyncio.run(
+            generate_tmp_text_file(self.test_text_content, self.user_id)
+        )
+
+        # Print the result for manual verification
+        print("\nTest Result:")
+        print(f"Generated storage path: {result}")
+
+        # Basic validation
+        if result.startswith("Error:"):
+            print(f"Test failed with error: {result}")
+            self.fail(f"Test failed with error: {result}")
+        else:
+            print("Test successful - storage path generated")
+            # Validate storage path format
+            expected_path_prefix = f"tmp/{self.user_id}/"
+            self.assertTrue(
+                result.startswith(expected_path_prefix),
+                f"Expected storage path to start with '{expected_path_prefix}', got: {result}",
+            )
+            
+            # Verify content by reading back the file
+            read_content = asyncio.run(read_tmp_text_file(result))
+            if read_content.startswith("Error:"):
+                self.fail(f"Failed to read back content: {read_content}")
+            else:
+                self.assertEqual(
+                    read_content, 
+                    self.test_text_content, 
+                    "Read content doesn't match original content"
+                )
+                print("Content verification successful - read content matches original")
+
+    def test_generate_tmp_text_file_with_unicode(self):
+        """Test generate_tmp_text_file function with unicode content"""
+        unicode_content = "Unicode test: 你好世界 🌍 ñáéíóú âêîôû àèìòù"
+
+        result = asyncio.run(
+            generate_tmp_text_file(unicode_content, self.user_id)
+        )
+
+        print(f"\nUnicode Test Result: {result}")
+
+        if result.startswith("Error:"):
+            print(f"Unicode test failed: {result}")
+            self.fail(f"Unicode test failed: {result}")
+        else:
+            print("Unicode test successful - storage path generated")
+            
+            # Verify content by reading back the file
+            read_content = asyncio.run(read_tmp_text_file(result))
+            if read_content.startswith("Error:"):
+                self.fail(f"Failed to read back unicode content: {read_content}")
+            else:
+                self.assertEqual(
+                    read_content, 
+                    unicode_content, 
+                    "Unicode read content doesn't match original content"
+                )
+                print("Unicode content verification successful - read content matches original")
+
+    def test_generate_tmp_text_file_with_empty_content(self):
+        """Test generate_tmp_text_file function with empty content"""
+        empty_content = ""
+
+        result = asyncio.run(
+            generate_tmp_text_file(empty_content, self.user_id)
+        )
+
+        print(f"\nEmpty Content Test Result: {result}")
+
+        if result.startswith("Error:"):
+            print(f"Empty content test failed: {result}")
+            self.fail(f"Empty content test failed: {result}")
+        else:
+            print("Empty content test successful - storage path generated")
+            
+            # Verify content by reading back the file
+            read_content = asyncio.run(read_tmp_text_file(result))
+            if read_content.startswith("Error:"):
+                self.fail(f"Failed to read back empty content: {read_content}")
+            else:
+                self.assertEqual(
+                    read_content, 
+                    empty_content, 
+                    "Empty read content doesn't match original content"
+                )
+                print("Empty content verification successful - read content matches original")
+
+
+if __name__ == "__main__":
+    unittest.main()

botrun_flow_lang/tests/test_mermaid_util.py

@@ -1,103 +1,103 @@
-import unittest
-import requests
-from botrun_flow_lang.langgraph_agents.agents.util.mermaid_util import (
-    generate_mermaid_files,
-)
-
-
-class TestMermaidUtil(unittest.TestCase):
-    def setUp(self):
-        """Set up test fixtures"""
-        self.botrun_flow_lang_url = (
-            "https://botrun-flow-lang-fastapi-dev-36186877499.asia-east1.run.app"
-        )
-        self.user_id = "sebastian.hsu@gmail.com"
-
-    def verify_mermaid_content(self, html_url: str):
-        """Helper method to verify the mermaid HTML content"""
-        # Verify URL format
-        self.assertTrue(html_url.endswith(".html"), "URL should end with .html")
-
-        # Get the HTML content
-        response = requests.get(html_url)
-        self.assertEqual(response.status_code, 200, "Failed to fetch HTML content")
-
-        html_content = response.text
-
-        # Check if it's a mermaid diagram
-        self.assertTrue(
-            "mermaid" in html_content, "HTML content should contain 'mermaid'"
-        )
-
-        # Check if it contains the required elements
-        self.assertTrue(
-            '<div class="mermaid"' in html_content,
-            "HTML content should contain mermaid div",
-        )
-        self.assertTrue(
-            "mermaid.min.js" in html_content,
-            "HTML content should contain mermaid.js script",
-        )
-
-    def test_generate_flowchart(self):
-        """Test generating a flowchart"""
-        # Test data for flowchart
-        flowchart_data = """
-        graph TD
-            A[開始] --> B{是否有資料?}
-            B -->|是| C[處理資料]
-            B -->|否| D[取得資料]
-            C --> E[結束]
-            D --> B
-        """
-
-        # Execute test
-        html_url = generate_mermaid_files(
-            mermaid_data=flowchart_data,
-            botrun_flow_lang_url=self.botrun_flow_lang_url,
-            user_id=self.user_id,
-            title="互動式流程圖",
-        )
-
-        # Verify results
-        self.verify_mermaid_content(html_url)
-
-    def test_generate_sequence_diagram(self):
-        """Test generating a sequence diagram"""
-        # Test data for sequence diagram
-        sequence_data = """
-        sequenceDiagram
-            participant 使用者
-            participant 系統
-            participant 資料庫
-            使用者->>系統: 請求資料
-            系統->>資料庫: 查詢資料
-            資料庫-->>系統: 返回結果
-            系統-->>使用者: 顯示資料
-        """
-
-        # Execute test
-        html_url = generate_mermaid_files(
-            mermaid_data=sequence_data,
-            botrun_flow_lang_url=self.botrun_flow_lang_url,
-            user_id=self.user_id,
-            title="互動式序列圖",
-        )
-
-        # Verify results
-        self.verify_mermaid_content(html_url)
-
-    def test_generate_mermaid_files_error(self):
-        """Test error handling with invalid data"""
-        result = generate_mermaid_files(
-            mermaid_data="invalid mermaid syntax",
-            botrun_flow_lang_url=self.botrun_flow_lang_url,
-            user_id=self.user_id,
-        )
-        self.assertFalse(
-            result.startswith("Error:")
-        )  # Mermaid will handle syntax errors client-side
-
-
-if __name__ == "__main__":
-    unittest.main()
+import unittest
+import requests
+from botrun_flow_lang.langgraph_agents.agents.util.mermaid_util import (
+    generate_mermaid_files,
+)
+
+
+class TestMermaidUtil(unittest.TestCase):
+    def setUp(self):
+        """Set up test fixtures"""
+        self.botrun_flow_lang_url = (
+            "https://botrun-flow-lang-fastapi-dev-36186877499.asia-east1.run.app"
+        )
+        self.user_id = "sebastian.hsu@gmail.com"
+
+    def verify_mermaid_content(self, html_url: str):
+        """Helper method to verify the mermaid HTML content"""
+        # Verify URL format
+        self.assertTrue(html_url.endswith(".html"), "URL should end with .html")
+
+        # Get the HTML content
+        response = requests.get(html_url)
+        self.assertEqual(response.status_code, 200, "Failed to fetch HTML content")
+
+        html_content = response.text
+
+        # Check if it's a mermaid diagram
+        self.assertTrue(
+            "mermaid" in html_content, "HTML content should contain 'mermaid'"
+        )
+
+        # Check if it contains the required elements
+        self.assertTrue(
+            '<div class="mermaid"' in html_content,
+            "HTML content should contain mermaid div",
+        )
+        self.assertTrue(
+            "mermaid.min.js" in html_content,
+            "HTML content should contain mermaid.js script",
+        )
+
+    def test_generate_flowchart(self):
+        """Test generating a flowchart"""
+        # Test data for flowchart
+        flowchart_data = """
+        graph TD
+            A[開始] --> B{是否有資料?}
+            B -->|是| C[處理資料]
+            B -->|否| D[取得資料]
+            C --> E[結束]
+            D --> B
+        """
+
+        # Execute test
+        html_url = generate_mermaid_files(
+            mermaid_data=flowchart_data,
+            botrun_flow_lang_url=self.botrun_flow_lang_url,
+            user_id=self.user_id,
+            title="互動式流程圖",
+        )
+
+        # Verify results
+        self.verify_mermaid_content(html_url)
+
+    def test_generate_sequence_diagram(self):
+        """Test generating a sequence diagram"""
+        # Test data for sequence diagram
+        sequence_data = """
+        sequenceDiagram
+            participant 使用者
+            participant 系統
+            participant 資料庫
+            使用者->>系統: 請求資料
+            系統->>資料庫: 查詢資料
+            資料庫-->>系統: 返回結果
+            系統-->>使用者: 顯示資料
+        """
+
+        # Execute test
+        html_url = generate_mermaid_files(
+            mermaid_data=sequence_data,
+            botrun_flow_lang_url=self.botrun_flow_lang_url,
+            user_id=self.user_id,
+            title="互動式序列圖",
+        )
+
+        # Verify results
+        self.verify_mermaid_content(html_url)
+
+    def test_generate_mermaid_files_error(self):
+        """Test error handling with invalid data"""
+        result = generate_mermaid_files(
+            mermaid_data="invalid mermaid syntax",
+            botrun_flow_lang_url=self.botrun_flow_lang_url,
+            user_id=self.user_id,
+        )
+        self.assertFalse(
+            result.startswith("Error:")
+        )  # Mermaid will handle syntax errors client-side
+
+
+if __name__ == "__main__":
+    unittest.main()

botrun_flow_lang/tests/test_pdf_analyzer.py

@@ -1,104 +1,104 @@
-import unittest
-import os
-from pathlib import Path
-from botrun_flow_lang.langgraph_agents.agents.util.local_files import (
-    upload_and_get_tmp_public_url,
-)
-from botrun_flow_lang.langgraph_agents.agents.util.pdf_analyzer import analyze_pdf
-
-
-class TestPDFAnalyzer(unittest.TestCase):
-    def setUp(self):
-        # Get the path to the test PDF file
-        current_dir = Path(__file__).parent
-        self.pdf_path = (
-            current_dir
-            / "test_files"
-            / "1120701A海廣離岸風力發電計畫環境影響說明書-C04.PDF"
-        )
-
-        # Ensure the test file exists
-        self.assertTrue(
-            os.path.exists(self.pdf_path), f"Test file not found at {self.pdf_path}"
-        )
-
-    def test_pdf_exists(self):
-        """Test if the PDF file exists"""
-        self.assertEqual(True, os.path.exists(self.pdf_path))
-
-    def test_pdf_readable(self):
-        """Test if the PDF file is readable"""
-        self.assertEqual(True, os.access(self.pdf_path, os.R_OK))
-
-    def test_pdf_not_empty(self):
-        """Test if the PDF file is not empty"""
-        self.assertEqual(True, os.path.getsize(self.pdf_path) > 0)
-
-    def test_pdf_extension(self):
-        """Test if the file has a PDF extension"""
-        self.assertEqual(True, self.pdf_path.suffix.lower() == ".pdf")
-
-    def test_analyze_table_4_3_1(self):
-        """Test analyzing table 4.3-1 from the PDF"""
-        query = "請你幫我找出在報告書中的「表 4.3-1 環境敏感地區調查表-第一級環境敏感地區」表格中的所有項目的「查詢結果及限制內容」幫我列出是或否？請全部列出來，不要遺漏"
-        pdf_url = upload_and_get_tmp_public_url(
-            self.pdf_path,
-            "https://botrun-flow-lang-fastapi-dev-36186877499.asia-east1.run.app",
-            "sebastian.hsu@gmail.com",
-        )
-        result = analyze_pdf(pdf_url, query)
-
-        # Print the result for inspection
-        print("\nAnalyze PDF Result:")
-        print("-" * 50)
-        print(result)
-        print("-" * 50)
-
-        # 確保結果不為空
-        self.assertIsNotNone(result)
-        self.assertNotEqual("", result)
-
-        # 檢查每個項目的結果
-        self.assertTrue(
-            "活動斷層兩側一定範圍: 否" in result
-            or "活動斷層兩側一定範圍：否" in result,
-        )
-        self.assertTrue(
-            "特定水土保持區: 否" in result or "特定水土保持區：否" in result
-        )
-        self.assertTrue("河川區域: 否" in result or "河川區域：否" in result)
-        self.assertTrue(
-            "洪氾區一級管制區及洪水平原一級管制區: 否" in result
-            or "洪氾區一級管制區及洪水平原一級管制區：否" in result
-        )
-        self.assertTrue(
-            "區域排水設施範圍: 是" in result or "區域排水設施範圍：是" in result
-        )
-        self.assertTrue(
-            "國家公園區內之特別景觀區、生態保護區: 否" in result
-            or "國家公園區內之特別景觀區、生態保護區：否" in result
-        )
-        self.assertTrue("自然保留區: 否" in result or "自然保留區：否" in result)
-        self.assertTrue(
-            "野生動物保護區: 否" in result or "野生動物保護區：否" in result
-        )
-        self.assertTrue(
-            "野生動物重要棲息環境: 是" in result or "野生動物重要棲息環境：是" in result
-        )
-        self.assertTrue("自然保護區: 否" in result or "自然保護區：否" in result)
-        self.assertTrue(
-            "一級海岸保護區: 是" in result or "一級海岸保護區：是" in result
-        )
-        self.assertTrue(
-            "國際級重要濕地、國家級重要濕地之核心保育區及生態復育區: 否" in result
-            or "國際級重要濕地、國家級重要濕地之核心保育區及生態復育區：否" in result
-        )
-        self.assertTrue("古蹟保存區: 否" in result or "古蹟保存區：否" in result)
-        self.assertTrue("考古遺址: 否" in result or "考古遺址：否" in result)
-        self.assertTrue(
-            "重要聚落建築群: 否" in result or "重要聚落建築群：否" in result
-        )
-
-
-if __name__ == "__main__":
-    unittest.main()
+import unittest
+import os
+from pathlib import Path
+from botrun_flow_lang.langgraph_agents.agents.util.local_files import (
+    upload_and_get_tmp_public_url,
+)
+from botrun_flow_lang.langgraph_agents.agents.util.pdf_analyzer import analyze_pdf
+
+
+class TestPDFAnalyzer(unittest.TestCase):
+    def setUp(self):
+        # Get the path to the test PDF file
+        current_dir = Path(__file__).parent
+        self.pdf_path = (
+            current_dir
+            / "test_files"
+            / "1120701A海廣離岸風力發電計畫環境影響說明書-C04.PDF"
+        )
+
+        # Ensure the test file exists
+        self.assertTrue(
+            os.path.exists(self.pdf_path), f"Test file not found at {self.pdf_path}"
+        )
+
+    def test_pdf_exists(self):
+        """Test if the PDF file exists"""
+        self.assertEqual(True, os.path.exists(self.pdf_path))
+
+    def test_pdf_readable(self):
+        """Test if the PDF file is readable"""
+        self.assertEqual(True, os.access(self.pdf_path, os.R_OK))
+
+    def test_pdf_not_empty(self):
+        """Test if the PDF file is not empty"""
+        self.assertEqual(True, os.path.getsize(self.pdf_path) > 0)
+
+    def test_pdf_extension(self):
+        """Test if the file has a PDF extension"""
+        self.assertEqual(True, self.pdf_path.suffix.lower() == ".pdf")
+
+    def test_analyze_table_4_3_1(self):
+        """Test analyzing table 4.3-1 from the PDF"""
+        query = "請你幫我找出在報告書中的「表 4.3-1 環境敏感地區調查表-第一級環境敏感地區」表格中的所有項目的「查詢結果及限制內容」幫我列出是或否？請全部列出來，不要遺漏"
+        pdf_url = upload_and_get_tmp_public_url(
+            self.pdf_path,
+            "https://botrun-flow-lang-fastapi-dev-36186877499.asia-east1.run.app",
+            "sebastian.hsu@gmail.com",
+        )
+        result = analyze_pdf(pdf_url, query)
+
+        # Print the result for inspection
+        print("\nAnalyze PDF Result:")
+        print("-" * 50)
+        print(result)
+        print("-" * 50)
+
+        # 確保結果不為空
+        self.assertIsNotNone(result)
+        self.assertNotEqual("", result)
+
+        # 檢查每個項目的結果
+        self.assertTrue(
+            "活動斷層兩側一定範圍: 否" in result
+            or "活動斷層兩側一定範圍：否" in result,
+        )
+        self.assertTrue(
+            "特定水土保持區: 否" in result or "特定水土保持區：否" in result
+        )
+        self.assertTrue("河川區域: 否" in result or "河川區域：否" in result)
+        self.assertTrue(
+            "洪氾區一級管制區及洪水平原一級管制區: 否" in result
+            or "洪氾區一級管制區及洪水平原一級管制區：否" in result
+        )
+        self.assertTrue(
+            "區域排水設施範圍: 是" in result or "區域排水設施範圍：是" in result
+        )
+        self.assertTrue(
+            "國家公園區內之特別景觀區、生態保護區: 否" in result
+            or "國家公園區內之特別景觀區、生態保護區：否" in result
+        )
+        self.assertTrue("自然保留區: 否" in result or "自然保留區：否" in result)
+        self.assertTrue(
+            "野生動物保護區: 否" in result or "野生動物保護區：否" in result
+        )
+        self.assertTrue(
+            "野生動物重要棲息環境: 是" in result or "野生動物重要棲息環境：是" in result
+        )
+        self.assertTrue("自然保護區: 否" in result or "自然保護區：否" in result)
+        self.assertTrue(
+            "一級海岸保護區: 是" in result or "一級海岸保護區：是" in result
+        )
+        self.assertTrue(
+            "國際級重要濕地、國家級重要濕地之核心保育區及生態復育區: 否" in result
+            or "國際級重要濕地、國家級重要濕地之核心保育區及生態復育區：否" in result
+        )
+        self.assertTrue("古蹟保存區: 否" in result or "古蹟保存區：否" in result)
+        self.assertTrue("考古遺址: 否" in result or "考古遺址：否" in result)
+        self.assertTrue(
+            "重要聚落建築群: 否" in result or "重要聚落建築群：否" in result
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()