update

tests/test_key_pattern_manager.py

@@ -0,0 +1,41 @@
+import unittest
+
+def validate_path():
+    import os, sys
+
+    os.path.dirname(__file__)
+    root_dir_assume = os.path.abspath(os.path.dirname(__file__) + "/..")
+    os.chdir(root_dir_assume)
+    sys.path.append(root_dir_assume)
+
+
+validate_path()  # validate path so you can run from base directory
+
+from shared_utils.key_pattern_manager import is_openai_api_key
+
+class TestKeyPatternManager(unittest.TestCase):
+    def test_is_openai_api_key_with_valid_key(self):
+        key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
+        self.assertTrue(is_openai_api_key(key))
+
+        key = "sx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
+        self.assertFalse(is_openai_api_key(key))
+
+        key = "sess-wg61ZafYHpNz7FFwIH7HGZlbVqUVaeV5tatHCWpl"
+        self.assertTrue(is_openai_api_key(key))
+
+        key = "sess-wg61ZafYHpNz7FFwIH7HGZlbVqUVa5tatHCWpl"
+        self.assertFalse(is_openai_api_key(key))
+
+
+    def test_is_openai_api_key_with_invalid_key(self):
+        key = "invalid_key"
+        self.assertFalse(is_openai_api_key(key))
+
+    def test_is_openai_api_key_with_custom_pattern(self):
+        # Assuming you have set a custom pattern in your configuration
+        key = "custom-pattern-key"
+        self.assertFalse(is_openai_api_key(key))
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_plugins.py

@@ -20,10 +20,10 @@ if __name__ == "__main__":
 
     # plugin_test(plugin='crazy_functions.函数动态生成->函数动态生成', main_input='交换图像的蓝色通道和红色通道', advanced_arg={"file_path_arg": "./build/ants.jpg"})
 
-    # plugin_test(plugin='crazy_functions.Latex输出PDF结果->Latex翻译中文并重新编译PDF', main_input="2307.07522")
+    # plugin_test(plugin='crazy_functions.Latex输出PDF->Latex翻译中文并重新编译PDF', main_input="2307.07522")
 
     plugin_test(
-        plugin="crazy_functions.Latex输出PDF结果->Latex翻译中文并重新编译PDF",
+        plugin="crazy_functions.Latex输出PDF->Latex翻译中文并重新编译PDF",
         main_input="G:/SEAFILE_LOCAL/50503047/我的资料库/学位/paperlatex/aaai/Fu_8368_with_appendix",
     )
 
@@ -66,7 +66,7 @@ if __name__ == "__main__":
 
     # plugin_test(plugin='crazy_functions.知识库文件注入->读取知识库作答', main_input="远程云服务器部署？")
 
-    # plugin_test(plugin='crazy_functions.Latex输出PDF结果->Latex翻译中文并重新编译PDF', main_input="2210.03629")
+    # plugin_test(plugin='crazy_functions.Latex输出PDF->Latex翻译中文并重新编译PDF', main_input="2210.03629")
 
     # advanced_arg = {"advanced_arg":"--llm_to_learn=gpt-3.5-turbo --prompt_prefix='根据下面的服装类型提示，想象一个穿着者，对这个人外貌、身处的环境、内心世界、人设进行描写。要求：100字以内，用第二人称。' --system_prompt=''" }
     # plugin_test(plugin='crazy_functions.chatglm微调工具->微调数据集生成', main_input='build/dev.json', advanced_arg=advanced_arg)