diff --git a/interpreter/core/llm/__init__.py b/interpreter/core/llm/__init__.py index e69de29bb2..d8472cd980 100644 --- a/interpreter/core/llm/__init__.py +++ b/interpreter/core/llm/__init__.py @@ -0,0 +1,14 @@ +from importlib import metadata + +_VERSION = "dev" + + +def __get_version(): + try: + version = metadata.version("open-interpreter") + except metadata.PackageNotFoundError: + version = _VERSION + return version + + +__version__ = __get_version() \ No newline at end of file diff --git a/interpreter/core/llm/llm.py b/interpreter/core/llm/llm.py index 980672db58..d3bbf66bea 100644 --- a/interpreter/core/llm/llm.py +++ b/interpreter/core/llm/llm.py @@ -4,6 +4,7 @@ import sys import litellm +from . import __version__ litellm.suppress_debug_info = True litellm.REPEATED_STREAMING_CHUNK_LIMIT = 99999999 @@ -437,6 +438,7 @@ def fixed_litellm_completions(**params): first_error = None params["num_retries"] = 0 + params["headers"] = {"x-coding-assistant": f"open-interpreter/{__version__}"} for attempt in range(attempts): try: