Ver Fonte

fix: truthy value (#8208)

Nam Vu há 7 meses atrás
pai
commit
342607f4a4

+ 1 - 1
api/core/callback_handler/index_tool_callback_handler.py

@@ -67,7 +67,7 @@ class DatasetIndexToolCallbackHandler:
                     data_source_type=item.get("data_source_type"),
                     segment_id=item.get("segment_id"),
                     score=item.get("score") if "score" in item else None,
-                    hit_count=item.get("hit_count") if "hit_count" else None,
+                    hit_count=item.get("hit_count") if "hit_count" in item else None,
                     word_count=item.get("word_count") if "word_count" in item else None,
                     segment_position=item.get("segment_position") if "segment_position" in item else None,
                     index_node_hash=item.get("index_node_hash") if "index_node_hash" in item else None,

+ 1 - 1
api/core/model_runtime/model_providers/vertex_ai/llm/llm.py

@@ -115,7 +115,7 @@ class VertexAiLargeLanguageModel(LargeLanguageModel):
             token = credentials.token
 
         # Vertex AI Anthropic Claude3 Opus model available in us-east5 region, Sonnet and Haiku available in us-central1 region
-        if "opus" or "claude-3-5-sonnet" in model:
+        if "opus" in model or "claude-3-5-sonnet" in model:
             location = "us-east5"
         else:
             location = "us-central1"