LLM-Bridge 1.15.4__tar.gz → 1.15.6__tar.gz

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/.github/workflows/python-publish.yml

@@ -1,3 +1,5 @@
+# https://docs.astral.sh/uv/guides/integration/github/
+
 name: "Publish"
 
 on:

llm_bridge-1.15.6/.github/workflows/python-test.yaml

@@ -0,0 +1,31 @@
+# https://docs.astral.sh/uv/guides/integration/github/
+
+name: Python test
+
+permissions:
+  contents: read
+
+on:
+  push:
+    branches: [ "main" ]
+  pull_request:
+    branches: [ "main" ]
+
+jobs:
+  uv:
+    name: python
+    runs-on: ubuntu-latest
+
+    steps:
+      - uses: actions/checkout@v5
+
+      - name: Enable caching
+        uses: astral-sh/setup-uv@v7
+        with:
+          enable-cache: true
+
+      - name: Install the project
+        run: uv sync --locked --all-extras --dev
+
+      - name: Run tests
+        run: uv run pytest tests

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: LLM-Bridge
-Version: 1.15.4
+Version: 1.15.6
 Summary: A Bridge for LLMs
 Author-email: windsnow1025 <windsnow1025@gmail.com>
 License-Expression: MIT

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/llm_bridge/logic/chat_generate/model_message_converter/openai_responses_message_converter.py

@@ -2,7 +2,7 @@ from openai.types.responses import ResponseInputTextParam, ResponseInputImagePar
     ResponseInputContentParam, EasyInputMessageParam, ResponseOutputMessageParam, ResponseInputFileParam
 
 from llm_bridge.logic.chat_generate import media_processor
-from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type, get_file_name
+from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type, get_filename_without_timestamp
 from llm_bridge.type.message import Message, ContentType
 from llm_bridge.type.model_message.openai_responses_message import OpenAIResponsesMessage
 
@@ -33,7 +33,7 @@ async def convert_message_to_openai_responses(message: Message) -> OpenAIRespons
                 file_data, _ = await media_processor.get_base64_content_from_url(file_url)
                 pdf_content = ResponseInputFileParam(
                     type="input_file",
-                    filename=get_file_name(file_url),
+                    filename=get_filename_without_timestamp(file_url),
                     file_data=f"data:application/pdf;base64,{file_data}",
                 )
                 content.append(pdf_content)

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/llm_bridge/logic/message_preprocess/file_type_checker.py

@@ -1,6 +1,7 @@
 import mimetypes
 import os
 import re
+from pathlib import PurePosixPath
 
 from llm_bridge.logic.file_fetch import fetch_file_data
 from llm_bridge.logic.message_preprocess.code_file_extensions import code_file_extensions
@@ -14,9 +15,12 @@ def is_file_type_supported(file_name: str) -> bool:
 
 
 async def get_file_type(file_url: str) -> tuple[str, str]:
-    file_name = get_file_name(file_url)
+    file_name = get_filename_without_timestamp(file_url)
+
+    # Treat filenames without an extension as their own extension
+    suffix = PurePosixPath(file_name).suffix.lower()
+    file_extension = suffix if suffix else '.' + file_name.lower()
 
-    file_extension = '.' + file_name.split('.')[-1].lower() # Treat filenames without an extension as their own extension
     if file_extension in code_file_extensions:
         return 'text', 'code'
     if file_extension == '.pdf':
@@ -41,9 +45,8 @@ async def get_file_type(file_url: str) -> tuple[str, str]:
     return 'unknown', 'unknown'
 
 
-# Without Timestamp
-def get_file_name(file_url: str) -> str:
-    base_name = os.path.basename(file_url)
+def get_filename_without_timestamp(file_url: str) -> str:
+    base_name = PurePosixPath(file_url).name
     match = re.search(r'-(.+)', base_name)
     if match:
         return match.group(1)

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/llm_bridge/logic/message_preprocess/message_preprocessor.py

@@ -1,5 +1,5 @@
 from llm_bridge.logic.message_preprocess import document_processor
-from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type, get_file_name
+from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type, get_filename_without_timestamp
 from llm_bridge.type.message import Message, Role, Content, ContentType
 
 
@@ -24,12 +24,12 @@ async def extract_text_files_to_message(message: Message, api_type: str) -> None
         if sub_type == "pdf" and api_type in ("OpenAI", "OpenAI-Azure", "Gemini-Vertex", "Gemini-Free", "Gemini-Paid", "Claude"):
             continue
             
-        filename = get_file_name(file_url)
+        filename = get_filename_without_timestamp(file_url)
         file_text = await document_processor.extract_text_from_file(file_url)
         
         message.contents[i] = Content(
             type=ContentType.Text,
-            data=f"{filename}: \n{file_text}\n"
+            data=f"<file name=\"{filename}\">\n{file_text}\n</file>"
         )
 
 

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/llm_bridge/logic/model_prices.py

@@ -1,5 +1,5 @@
-import importlib.resources
 import json
+from importlib.resources import files
 from typing import TypedDict
 
 from fastapi import HTTPException
@@ -12,13 +12,17 @@ class ModelPrice(TypedDict):
     output: float
 
 
-def load_json_file(package, filename):
-    with importlib.resources.open_text(package, filename) as f:
-        return json.load(f)
+def load_json_file(package: str, filename: str):
+    content = files(package).joinpath(filename).read_text(encoding="utf-8")
+    return json.loads(content)
 
 
 def get_model_prices() -> list[ModelPrice]:
-    return load_json_file("llm_bridge.resources", "model_prices.json")
+    prices = load_json_file("llm_bridge.resources", "model_prices.json")
+    for price in prices:
+        price["input"] = float(price["input"])
+        price["output"] = float(price["output"])
+    return prices
 
 
 def find_model_prices(api_type: str, model: str) -> ModelPrice | None:

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "LLM-Bridge"
-version = "1.15.4"
+version = "1.15.6"
 dependencies = [
     "fastapi",
     "httpx",

llm_bridge-1.15.6/tests/file_type_checker_test.py

@@ -0,0 +1,17 @@
+import pytest
+
+from llm_bridge.logic.message_preprocess.file_type_checker import get_file_type
+
+
+@pytest.mark.asyncio
+async def test_get_file_type_with_extension():
+    file_type, sub_type = await get_file_type("https://example.com/1767243600000-document.pdf")
+    assert file_type == "text"
+    assert sub_type == "pdf"
+
+
+@pytest.mark.asyncio
+async def test_get_file_type_without_extension():
+    file_type, sub_type = await get_file_type("https://example.com/1767243600000-Dockerfile")
+    assert file_type == "text"
+    assert sub_type == "code"

llm_bridge-1.15.6/tests/model_prices_test.py

@@ -0,0 +1,10 @@
+from llm_bridge.logic.model_prices import get_model_prices
+
+
+def test_get_model_prices_types():
+    result = get_model_prices()
+    for model_price in result:
+        assert isinstance(model_price["apiType"], str)
+        assert isinstance(model_price["model"], str)
+        assert isinstance(model_price["input"], float)
+        assert isinstance(model_price["output"], float)

llm_bridge-1.15.6/usage/config.py

@@ -0,0 +1,129 @@
+from llm_bridge import *
+
+structured_output_schema = {
+  "$schema": "https://json-schema.org/draft/2020-12/schema",
+  "$id": "https://example.com/product.schema.json",
+  "title": "Product",
+  "description": "A product from Acme's catalog",
+  "type": "object",
+  "properties": {
+    "productId": {
+      "description": "The unique identifier for a product",
+      "type": "integer"
+    },
+    "productName": {
+      "description": "Name of the product",
+      "type": "string"
+    },
+    "price": {
+      "description": "The price of the product",
+      "type": "number",
+      "exclusiveMinimum": 0
+    },
+    "tags": {
+      "description": "Tags for the product",
+      "type": "array",
+      "items": {
+        "type": "string"
+      },
+      "minItems": 1,
+      "uniqueItems": True
+    }
+  },
+  "required": [
+    "productId",
+    "productName",
+    "price"
+  ]
+}
+structured_output_schema = None
+
+messages = [
+    Message(
+        role=Role.System,
+        contents=[
+            Content(type=ContentType.Text, data="You are a helpful assistant.")
+        ]
+    ),
+    Message(
+        role=Role.User,
+        contents=[
+            Content(type=ContentType.Text, data="Hello")
+        ]
+    ),
+    Message(
+        role=Role.Assistant,
+        contents=[
+            Content(type=ContentType.Text, data="Hello! How can I assist you today?")
+        ]
+    ),
+    # Message(
+    #     role=Role.User,
+    #     contents=[
+    #         # Thinking
+    #         # Content(type=ContentType.Text, data="Explain the concept of Occam's Razor and provide a simple, everyday example."),
+    #
+    #         # Web Search
+    #         # Content(type=ContentType.Text, data="What's the weather in NYC today?"),
+    #
+    #         # Image Understanding
+    #         # Content(type=ContentType.File, data="https://www.gstatic.com/webp/gallery3/1.png"),
+    #         # Content(type=ContentType.Text, data="What is in this image?"),
+    #
+    #         # Image Generation
+    #         # Content(type=ContentType.Text, data="Please generate an image of a cat."),
+    #
+    #         # URL Context
+    #         # Content(type=ContentType.Text, data="What is in https://www.windsnow1025.com/"),
+    #
+    #         # Code Execution
+    #         # Content(type=ContentType.Text, data="What is the sum of the first 50 prime numbers? Generate and run code for the calculation, and make sure you get all 50."),
+    #
+    #         # File Output
+    #         # Content(type=ContentType.Text, data="Create a matplotlib visualization and save it as output.png"),
+    #
+    #         # Structured Output
+    #         # Content(type=ContentType.Text, data="Please generate a product."),
+    #     ]
+    # ),
+    Message(
+        role=Role.User,
+        contents=[
+            Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1769429581512-Test.txt"),
+            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746208707489-image.png"),
+            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746209841847-A%20Tutorial%20on%20Spectral%20Clustering.pdf"),
+            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212253473-file_example_MP3_700KB.mp3"),
+            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212980820-file_example_MP4_480_1_5MG.mp4"),
+            # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1753804900037-Calculus.docx"),
+            Content(type=ContentType.Text, data="What's this?"),
+        ]
+    ),
+]
+# See /llm_bridge/resources/model_prices.json for available models
+# model = "gpt-5.2"
+# model = "gpt-5.1"
+# model = "gpt-5-pro"
+# model = "gpt-5"
+# model = "gpt-4.1"
+# model = "gpt-5-codex"
+# model = "gemini-3-pro-preview"
+# model = "gemini-3-pro-image-preview"
+model = "gemini-3-flash-preview"
+# model = "grok-4-1-fast-reasoning"
+# model = "claude-sonnet-4-5"
+# model = "claude-opus-4-5"
+api_type = "Gemini-Vertex"
+# api_type = "Gemini-Free"
+# api_type = "Gemini-Paid"
+# api_type = "OpenAI"
+# api_type = "OpenAI-Azure"
+# api_type = "OpenAI-GitHub"
+# api_type = "Claude"
+# api_type = "Grok"
+temperature = 0
+stream = True
+# stream = False
+thought = True
+# thought = False
+code_execution = True
+# code_execution = False

llm_bridge-1.15.6/usage/main.py

@@ -0,0 +1,100 @@
+import asyncio
+import logging
+import os
+import sys
+from pathlib import Path
+from pprint import pprint
+
+from dotenv import load_dotenv
+
+from usage.config import *
+from usage.workflow import workflow
+
+script_dir = Path(__file__).parent.resolve()
+
+# Env
+load_dotenv(script_dir / ".env")
+
+# Logging Output File
+output_path = script_dir / "output.log"
+output_path.parent.mkdir(parents=True, exist_ok=True)
+output_file = output_path.open("w", encoding="utf-8")
+sys.stdout = output_file
+
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(levelname)s - %(message)s',
+    stream=output_file
+)
+
+api_keys = {
+    "OPENAI_API_KEY": os.environ.get("OPENAI_API_KEY"),
+    "AZURE_API_KEY": os.environ.get("AZURE_API_KEY"),
+    "AZURE_API_BASE": os.environ.get("AZURE_API_BASE"),
+    "GITHUB_API_KEY": os.environ.get("GITHUB_API_KEY"),
+    "GEMINI_FREE_API_KEY": os.environ.get("GEMINI_FREE_API_KEY"),
+    "GEMINI_PAID_API_KEY": os.environ.get("GEMINI_PAID_API_KEY"),
+    "GEMINI_VERTEX_API_KEY": os.environ.get("GEMINI_VERTEX_API_KEY"),
+    "ANTHROPIC_API_KEY": os.environ.get("ANTHROPIC_API_KEY"),
+    "XAI_API_KEY": os.environ.get("XAI_API_KEY"),
+}
+
+
+async def main():
+    input_tokens = 0
+    output_tokens = 0
+    response = await workflow(
+        api_keys,
+        messages,
+        model,
+        api_type,
+        temperature,
+        stream,
+        thought,
+        code_execution,
+        structured_output_schema,
+    )
+    text = ""
+    thought_text = ""
+    code_text = ""
+    code_output_text = ""
+    files = []
+
+    if stream:
+        async for chunk in response:
+            pprint(chunk)
+            if chunk.text:
+                text += chunk.text
+            if chunk.thought:
+                thought_text += chunk.thought
+            if chunk.input_tokens:
+                input_tokens = chunk.input_tokens
+            if chunk.output_tokens:
+                output_tokens += chunk.output_tokens
+            if chunk.code:
+                code_text += chunk.code
+            if chunk.code_output:
+                code_output_text += chunk.code_output
+            if chunk.files:
+                files.extend(chunk.files)
+    else:
+        pprint(response)
+        text = response.text
+        thought_text = response.thought
+        code_text = response.code
+        code_output_text = response.code_output
+        input_tokens = response.input_tokens
+        output_tokens = response.output_tokens
+        files = response.files
+    total_cost = calculate_chat_cost(api_type, model, input_tokens, output_tokens)
+    print(f"Thought:\n{thought_text}\n")
+    print(f"Code:\n{code_text}\n")
+    print(f"Code Output:\n{code_output_text}\n")
+    print(f"Text:\n{text}\n")
+    print(f"Files:\n{files}\n")
+    print(f'Input tokens: {input_tokens}, Output tokens: {output_tokens}, Total cost: ${total_cost}')
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+    output_file.close()

{llm_bridge-1.15.4 → llm_bridge-1.15.6}/uv.lock

@@ -357,7 +357,7 @@ wheels = [
 
 [[package]]
 name = "llm-bridge"
-version = "1.15.4"
+version = "1.15.6"
 source = { editable = "." }
 dependencies = [
     { name = "anthropic" },

llm_bridge-1.15.4/tests/chat_client_factory_test.py

@@ -1,20 +0,0 @@
-import pytest
-
-from llm_bridge.type.message import Message, Role, Content, ContentType
-
-
-@pytest.fixture
-def sample_messages():
-    return [
-        Message(role=Role.System, contents=[
-            Content(type=ContentType.Text, data="You are a helpful assistant.")
-        ]),
-        Message(role=Role.User, contents=[
-            Content(type=ContentType.Text, data="Hello")
-        ])
-    ]
-
-
-@pytest.mark.asyncio
-async def test_placeholder():
-    assert True

llm_bridge-1.15.4/tests/message_preprocessor_test.py

@@ -1,26 +0,0 @@
-import pytest
-
-from llm_bridge.logic.message_preprocess.message_preprocessor import extract_system_messages
-from llm_bridge.type.message import Message, Role, Content, ContentType
-
-
-@pytest.fixture
-def sample_messages():
-    return [
-        Message(role=Role.System, contents=[
-            Content(type=ContentType.Text, data="You are a helpful assistant.")
-        ]),
-        Message(role=Role.User, contents=[
-            Content(type=ContentType.Text, data="Hello")
-        ])
-    ]
-
-def test_extract_system_messages(sample_messages):
-    extracted_text = extract_system_messages(sample_messages)
-
-    assert extracted_text == "You are a helpful assistant.\n"
-
-    assert len(sample_messages) == 1
-    assert sample_messages[0].role == Role.User
-    assert sample_messages[0].contents[0].type == ContentType.Text
-    assert sample_messages[0].contents[0].data == "Hello"

llm_bridge-1.15.4/usage/main.py

@@ -1,226 +0,0 @@
-import asyncio
-import logging
-import os
-import sys
-from pprint import pprint
-
-from dotenv import load_dotenv
-
-from llm_bridge import *
-from usage.workflow import workflow
-
-output_file = open("./usage/output.log", "w", encoding="utf-8")
-sys.stdout = output_file
-
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    stream=output_file
-)
-
-script_dir = os.path.dirname(os.path.abspath(__file__))
-env_path = os.path.join(script_dir, ".env")
-load_dotenv(env_path)
-
-api_keys = {
-    "OPENAI_API_KEY": os.environ.get("OPENAI_API_KEY"),
-    "AZURE_API_KEY": os.environ.get("AZURE_API_KEY"),
-    "AZURE_API_BASE": os.environ.get("AZURE_API_BASE"),
-    "GITHUB_API_KEY": os.environ.get("GITHUB_API_KEY"),
-    "GEMINI_FREE_API_KEY": os.environ.get("GEMINI_FREE_API_KEY"),
-    "GEMINI_PAID_API_KEY": os.environ.get("GEMINI_PAID_API_KEY"),
-    "GEMINI_VERTEX_API_KEY": os.environ.get("GEMINI_VERTEX_API_KEY"),
-    "ANTHROPIC_API_KEY": os.environ.get("ANTHROPIC_API_KEY"),
-    "XAI_API_KEY": os.environ.get("XAI_API_KEY"),
-}
-
-structured_output_schema = {
-  "$schema": "https://json-schema.org/draft/2020-12/schema",
-  "$id": "https://example.com/product.schema.json",
-  "title": "Product",
-  "description": "A product from Acme's catalog",
-  "type": "object",
-  "properties": {
-    "productId": {
-      "description": "The unique identifier for a product",
-      "type": "integer"
-    },
-    "productName": {
-      "description": "Name of the product",
-      "type": "string"
-    },
-    "price": {
-      "description": "The price of the product",
-      "type": "number",
-      "exclusiveMinimum": 0
-    },
-    "tags": {
-      "description": "Tags for the product",
-      "type": "array",
-      "items": {
-        "type": "string"
-      },
-      "minItems": 1,
-      "uniqueItems": True
-    }
-  },
-  "required": [
-    "productId",
-    "productName",
-    "price"
-  ]
-}
-structured_output_schema = None
-
-messages = [
-    Message(
-        role=Role.System,
-        contents=[
-            Content(type=ContentType.Text, data="You are a helpful assistant.")
-        ]
-    ),
-    Message(
-        role=Role.User,
-        contents=[
-            Content(type=ContentType.Text, data="Hello")
-        ]
-    ),
-    Message(
-        role=Role.Assistant,
-        contents=[
-            Content(type=ContentType.Text, data="Hello! How can I assist you today?")
-        ]
-    ),
-    Message(
-        role=Role.User,
-        contents=[
-            # Thinking
-            # Content(type=ContentType.Text, data="Explain the concept of Occam's Razor and provide a simple, everyday example."),
-
-            # Web Search
-            # Content(type=ContentType.Text, data="What's the weather in NYC today?"),
-
-            # Image Understanding
-            # Content(type=ContentType.File, data="https://www.gstatic.com/webp/gallery3/1.png"),
-            # Content(type=ContentType.Text, data="What is in this image?"),
-
-            # Image Generation
-            # Content(type=ContentType.Text, data="Please generate an image of a cat."),
-
-            # URL Context
-            # Content(type=ContentType.Text, data="What is in https://www.windsnow1025.com/"),
-
-            # Code Execution
-            # Content(type=ContentType.Text, data="What is the sum of the first 50 prime numbers? Generate and run code for the calculation, and make sure you get all 50."),
-
-            # File Output
-            Content(type=ContentType.Text, data="Create a matplotlib visualization and save it as output.png"),
-
-            # Structured Output
-            # Content(type=ContentType.Text, data="Please generate a product."),
-        ]
-    ),
-    # Message(
-    #     role=Role.User,
-    #     contents=[
-    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746208707489-image.png"),
-    #         Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746209841847-A%20Tutorial%20on%20Spectral%20Clustering.pdf"),
-    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212253473-file_example_MP3_700KB.mp3"),
-    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1746212980820-file_example_MP4_480_1_5MG.mp4"),
-    #         # Content(type=ContentType.File, data="https://www.windsnow1025.com/minio/windsnow/uploads/1/1753804900037-Calculus.docx"),
-    #         Content(type=ContentType.Text, data="What's this?"),
-    #     ]
-    # ),
-]
-# See /llm_bridge/resources/model_prices.json for available models
-model = "gpt-5.2"
-# model = "gpt-5.1"
-# model = "gpt-5-pro"
-# model = "gpt-5"
-# model = "gpt-4.1"
-# model = "gpt-5-codex"
-# model = "gemini-3-pro-preview"
-# model = "gemini-3-pro-image-preview"
-# model = "gemini-3-flash-preview"
-# model = "grok-4-1-fast-reasoning"
-# model = "claude-sonnet-4-5"
-# model = "claude-opus-4-5"
-# api_type = "Gemini-Vertex"
-# api_type = "Gemini-Free"
-# api_type = "Gemini-Paid"
-api_type = "OpenAI"
-# api_type = "OpenAI-Azure"
-# api_type = "OpenAI-GitHub"
-# api_type = "Claude"
-# api_type = "Grok"
-temperature = 0
-stream = True
-# stream = False
-thought = True
-# thought = False
-code_execution = True
-# code_execution = False
-
-
-async def main():
-    model_prices = get_model_prices()
-    pprint(model_prices)
-    print(structured_output_schema)
-
-    input_tokens = 0
-    output_tokens = 0
-    response = await workflow(
-        api_keys,
-        messages,
-        model,
-        api_type,
-        temperature,
-        stream,
-        thought,
-        code_execution,
-        structured_output_schema,
-    )
-    text = ""
-    thought_text = ""
-    code_text = ""
-    code_output_text = ""
-    files = []
-
-    if stream:
-        async for chunk in response:
-            pprint(chunk)
-            if chunk.text:
-                text += chunk.text
-            if chunk.thought:
-                thought_text += chunk.thought
-            if chunk.input_tokens:
-                input_tokens = chunk.input_tokens
-            if chunk.output_tokens:
-                output_tokens += chunk.output_tokens
-            if chunk.code:
-                code_text += chunk.code
-            if chunk.code_output:
-                code_output_text += chunk.code_output
-            if chunk.files:
-                files.extend(chunk.files)
-    else:
-        pprint(response)
-        text = response.text
-        thought_text = response.thought
-        code_text = response.code
-        code_output_text = response.code_output
-        input_tokens = response.input_tokens
-        output_tokens = response.output_tokens
-        files = response.files
-    total_cost = calculate_chat_cost(api_type, model, input_tokens, output_tokens)
-    print(f"Thought:\n{thought_text}\n")
-    print(f"Code:\n{code_text}\n")
-    print(f"Code Output:\n{code_output_text}\n")
-    print(f"Text:\n{text}\n")
-    print(f"Files:\n{files}\n")
-    print(f'Input tokens: {input_tokens}, Output tokens: {output_tokens}, Total cost: ${total_cost}')
-
-
-if __name__ == "__main__":
-    asyncio.run(main())
-    output_file.close()