pycityagent 2.0.0a93__cp311-cp311-macosx_11_0_arm64.whl → 2.0.0a94__cp311-cp311-macosx_11_0_arm64.whl

pycityagent/llm/llm.py

@@ -44,12 +44,12 @@ class LLM:
             - `config`: An instance of `LLMConfig` containing configuration settings for the LLM.
         """
         self.config = config
-        if config.text["request_type"] not in ["openai", "deepseek", "qwen", "zhipuai"]:
+        if config.text["request_type"] not in ["openai", "deepseek", "qwen", "zhipuai", "siliconflow"]:
             raise ValueError("Invalid request type for text request")
         self.prompt_tokens_used = 0
         self.completion_tokens_used = 0
         self.request_number = 0
-        self.semaphore = None
+        self.semaphore = asyncio.Semaphore(200)
         self._current_client_index = 0
         self._log_list = []
 
@@ -75,6 +75,12 @@ class LLM:
                     base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
                     timeout=300,
                 )
+            elif self.config.text["request_type"] == "siliconflow":
+                client = AsyncOpenAI(
+                    api_key=api_key,
+                    base_url="https://api.siliconflow.cn/v1",
+                    timeout=300,
+                )
             elif self.config.text["request_type"] == "zhipuai":
                 client = ZhipuAI(api_key=api_key, timeout=300)
             else:
@@ -237,6 +243,7 @@ class LLM:
                 self.config.text["request_type"] == "openai"
                 or self.config.text["request_type"] == "deepseek"
                 or self.config.text["request_type"] == "qwen"
+                or self.config.text["request_type"] == "siliconflow"
             ):
                 for attempt in range(retries):
                     try:

{pycityagent-2.0.0a93.dist-info → pycityagent-2.0.0a94.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycityagent
-Version: 2.0.0a93
+Version: 2.0.0a94
 Summary: LLM-based city environment agent building library
 Author-email: Yuwei Yan <pinkgranite86@gmail.com>, Junbo Yan <yanjb20thu@gmali.com>, Jun Zhang <zhangjun990222@gmali.com>
 License: MIT License

{pycityagent-2.0.0a93.dist-info → pycityagent-2.0.0a94.dist-info}/RECORD

@@ -9,7 +9,7 @@ pycityagent/tools/__init__.py,sha256=y7sMVMHf0AbivlczM2h-kr7mkgXK-WAx3S9BXLXkWvw
 pycityagent/tools/tool.py,sha256=4ZJSHbNM8dfAVwZEw8T0a2_9OuPsPQpKSVL4WxZVBUc,9022
 pycityagent/llm/llmconfig.py,sha256=6AqCMV4B_JqBD2mb98bLGzpUdlOCnziQKae-Hhxxp-E,469
 pycityagent/llm/__init__.py,sha256=iWs6FLgrbRVIiqOf4ILS89gkVCTvS7HFC3vG-MWuyko,205
-pycityagent/llm/llm.py,sha256=YJy5SleUpHvS5v0S-wnnrY4zTbu_2RL36LH-zCYKPEI,20048
+pycityagent/llm/llm.py,sha256=FgtRxpp53scmUH9i7XwaTXZ-bayw-bAanXzlRZuY6wY,20406
 pycityagent/llm/embeddings.py,sha256=3610I-_scAy8HwRNpT8hVJpH9_8_pTLCPptqnzSq10o,11322
 pycityagent/llm/utils.py,sha256=rSx_fp-_Gh0vZ-x2rqAUqnpS56BVTZ4ChfAMarB8S1A,195
 pycityagent/memory/memory.py,sha256=5mUweo-BgQYbXmVRAk7HTUXsar6O6iYz79VbLuMAvjo,45063
@@ -89,9 +89,9 @@ pycityagent/cityagent/blocks/mobility_block.py,sha256=OZoA-swFdX0Ezpzkyp8jJMNPQ5
 pycityagent/survey/models.py,sha256=g3xni4GcA1Py3vlGt6z4ltutjgQ4G0uINYAM8vKRJAw,5225
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=tHkdeq4lTfAHwvgf4-udsXri0z2l6E00rEbvwl7SqRs,3439
-pycityagent-2.0.0a93.dist-info/RECORD,,
-pycityagent-2.0.0a93.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
-pycityagent-2.0.0a93.dist-info/WHEEL,sha256=NW1RskY9zow1Y68W-gXg0oZyBRAugI1JHywIzAIai5o,109
-pycityagent-2.0.0a93.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
-pycityagent-2.0.0a93.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
-pycityagent-2.0.0a93.dist-info/METADATA,sha256=p5FYOWipjHs7S50Kd0R-kNN5NB-vWfLgDbyIzFLYW0Q,9110
+pycityagent-2.0.0a94.dist-info/RECORD,,
+pycityagent-2.0.0a94.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
+pycityagent-2.0.0a94.dist-info/WHEEL,sha256=NW1RskY9zow1Y68W-gXg0oZyBRAugI1JHywIzAIai5o,109
+pycityagent-2.0.0a94.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
+pycityagent-2.0.0a94.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
+pycityagent-2.0.0a94.dist-info/METADATA,sha256=oHBYTha8EtA0rjMUS4UN4_pYb9BkkoVP4ZYX75t9DgQ,9110