Fixing syntax error from mis-merge

This commit is contained in:
Luke
2023-06-19 19:46:46 +00:00
parent 71ca4ea990
commit 2a8f4ce0b7

View File

@@ -4,14 +4,8 @@ from typing import List, Optional
import openai
from openai import Model
<<<<<<< HEAD
from autogpt.config import Config
from autogpt.llm.base import CompletionModelInfo, MessageDict
from autogpt.llm.base import CompletionModelInfo
from autogpt.llm.providers.openai import OPEN_AI_MODELS
=======
from autogpt.llm.modelsinfo import COSTS
>>>>>>> upstream/master
from autogpt.logs import logger
from autogpt.singleton import Singleton
@@ -30,8 +24,8 @@ class ApiManager(metaclass=Singleton):
self.total_cost = 0
self.total_budget = 0.0
self.models = None
def update_cost(self, prompt_tokens, completion_tokens, model)
def update_cost(self, prompt_tokens, completion_tokens, model):
"""
Update the total cost, prompt tokens, and completion tokens.