File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
homeassistant/components/google_generative_ai_conversation Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change 1919 CONF_CHAT_MODEL ,
2020 CONF_RECOMMENDED ,
2121 LOGGER ,
22- RECOMMENDED_A_TASK_MAX_TOKENS ,
22+ RECOMMENDED_AI_TASK_MAX_TOKENS ,
2323 RECOMMENDED_IMAGE_MODEL ,
2424)
2525from .entity import (
@@ -80,7 +80,7 @@ async def _async_generate_data(
8080 ) -> ai_task .GenDataTaskResult :
8181 """Handle a generate data task."""
8282 await self ._async_handle_chat_log (
83- chat_log , task .structure , default_max_tokens = RECOMMENDED_A_TASK_MAX_TOKENS
83+ chat_log , task .structure , default_max_tokens = RECOMMENDED_AI_TASK_MAX_TOKENS
8484 )
8585
8686 if not isinstance (chat_log .content [- 1 ], conversation .AssistantContent ):
Original file line number Diff line number Diff line change 3333CONF_MAX_TOKENS = "max_tokens"
3434RECOMMENDED_MAX_TOKENS = 3000
3535# Input 5000, output 19400 = 0.05 USD
36- RECOMMENDED_A_TASK_MAX_TOKENS = 19400
36+ RECOMMENDED_AI_TASK_MAX_TOKENS = 19400
3737CONF_HARASSMENT_BLOCK_THRESHOLD = "harassment_block_threshold"
3838CONF_HATE_BLOCK_THRESHOLD = "hate_block_threshold"
3939CONF_SEXUAL_BLOCK_THRESHOLD = "sexual_block_threshold"
You can’t perform that action at this time.
0 commit comments