promptbuilder 0.4.25__tar.gz → 0.4.27__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (35) hide show
  1. {promptbuilder-0.4.25/promptbuilder.egg-info → promptbuilder-0.4.27}/PKG-INFO +1 -1
  2. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/base_client.py +6 -2
  3. {promptbuilder-0.4.25 → promptbuilder-0.4.27/promptbuilder.egg-info}/PKG-INFO +1 -1
  4. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/setup.py +1 -1
  5. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/LICENSE +0 -0
  6. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/MANIFEST.in +0 -0
  7. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/Readme.md +0 -0
  8. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/__init__.py +0 -0
  9. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/agent/__init__.py +0 -0
  10. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/agent/agent.py +0 -0
  11. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/agent/context.py +0 -0
  12. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/agent/tool.py +0 -0
  13. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/agent/utils.py +0 -0
  14. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/embeddings.py +0 -0
  15. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/__init__.py +0 -0
  16. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/aisuite_client.py +0 -0
  17. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/anthropic_client.py +0 -0
  18. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/bedrock_client.py +0 -0
  19. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/config.py +0 -0
  20. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/exceptions.py +0 -0
  21. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/google_client.py +0 -0
  22. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/logfire_decorators.py +0 -0
  23. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/main.py +0 -0
  24. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/openai_client.py +0 -0
  25. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/types.py +0 -0
  26. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/llm_client/utils.py +0 -0
  27. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder/prompt_builder.py +0 -0
  28. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder.egg-info/SOURCES.txt +0 -0
  29. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder.egg-info/dependency_links.txt +0 -0
  30. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder.egg-info/requires.txt +0 -0
  31. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/promptbuilder.egg-info/top_level.txt +0 -0
  32. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/pyproject.toml +0 -0
  33. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/setup.cfg +0 -0
  34. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/tests/test_llm_client.py +0 -0
  35. {promptbuilder-0.4.25 → promptbuilder-0.4.27}/tests/test_llm_client_async.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: promptbuilder
3
- Version: 0.4.25
3
+ Version: 0.4.27
4
4
  Summary: Library for building prompts for LLMs
5
5
  Home-page: https://github.com/kapulkin/promptbuilder
6
6
  Author: Kapulkin Stanislav
@@ -104,7 +104,8 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
104
104
 
105
105
  total_count = BaseLLMClient._response_out_tokens(response)
106
106
 
107
- while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
107
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
108
+ while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
108
109
  BaseLLMClient._append_generated_part(messages, response)
109
110
 
110
111
  response = self._create(
@@ -116,6 +117,7 @@ class BaseLLMClient(ABC, utils.InheritDecoratorsMixin):
116
117
  tools=tools,
117
118
  tool_config=tool_config,
118
119
  )
120
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
119
121
  total_count += BaseLLMClient._response_out_tokens(response)
120
122
  if max_tokens is not None and total_count >= max_tokens:
121
123
  break
@@ -436,7 +438,8 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
436
438
 
437
439
  total_count = BaseLLMClient._response_out_tokens(response)
438
440
 
439
- while autocomplete and response.candidates and response.candidates[0].finish_reason == FinishReason.MAX_TOKENS:
441
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
442
+ while autocomplete and response.candidates and finish_reason == FinishReason.MAX_TOKENS.value:
440
443
  BaseLLMClient._append_generated_part(messages, response)
441
444
 
442
445
  response = await self._create(
@@ -448,6 +451,7 @@ class BaseLLMClientAsync(ABC, utils.InheritDecoratorsMixin):
448
451
  tools=tools,
449
452
  tool_config=tool_config,
450
453
  )
454
+ finish_reason = response.candidates[0].finish_reason.value if response.candidates and response.candidates[0].finish_reason else None
451
455
  total_count += BaseLLMClient._response_out_tokens(response)
452
456
  if max_tokens is not None and total_count >= max_tokens:
453
457
  break
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: promptbuilder
3
- Version: 0.4.25
3
+ Version: 0.4.27
4
4
  Summary: Library for building prompts for LLMs
5
5
  Home-page: https://github.com/kapulkin/promptbuilder
6
6
  Author: Kapulkin Stanislav
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
2
2
 
3
3
  setup(
4
4
  name="promptbuilder",
5
- version="0.4.25",
5
+ version="0.4.27",
6
6
  packages=find_packages(),
7
7
  install_requires=[
8
8
  "pydantic",
File without changes
File without changes
File without changes