From 7f5f2336bc46cd8df486b8f22664e7a5073b2a6d Mon Sep 17 00:00:00 2001
From: Richard Blythman <richardblythman@gmail.com>
Date: Wed, 27 Mar 2024 15:36:18 +0000
Subject: [PATCH] add older model prices to counter for polywrap tools

---
 packages/valory/skills/task_execution/utils/benchmarks.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/packages/valory/skills/task_execution/utils/benchmarks.py b/packages/valory/skills/task_execution/utils/benchmarks.py
index 58ddd1f2..224a0381 100644
--- a/packages/valory/skills/task_execution/utils/benchmarks.py
+++ b/packages/valory/skills/task_execution/utils/benchmarks.py
@@ -31,9 +31,11 @@ class TokenCounterCallback:
     TOKEN_PRICES = {
         "gpt-3.5-turbo": {"input": 0.0005, "output": 0.0015},
         "gpt-3.5-turbo-0125": {"input": 0.0005, "output": 0.0015},
+        "gpt-3.5-turbo-1106": {"input": 0.001, "output": 0.002},
         "gpt-4": {"input": 0.03, "output": 0.06},
         "gpt-4-turbo-preview": {"input": 0.01, "output": 0.03},
         "gpt-4-0125-preview": {"input": 0.01, "output": 0.03},
+        "gpt-4-1106-preview": {"input": 0.01, "output": 0.03},
         "claude-2": {"input": 0.008, "output": 0.024},
     }