Add missing imports to ai_providers.py

This commit is contained in:
2024-08-18 13:42:14 -05:00
parent bcdb6cf0d9
commit be7d9c7817
+4 -2
View File
@@ -1,8 +1,10 @@
import os
import json
import logging
import anthropic
from openai import OpenAI
from abc import ABC, abstractmethod
from functions import functions
class AIProvider(ABC):
@abstractmethod
@@ -56,7 +58,7 @@ class AnthropicProvider(AIProvider):
"description": function['description'],
"input_schema": function['parameters'] if function['parameters'] not in [None, {}] else {"type": "object", "properties": {"param1": {"type": "string", "description": "Unnecessary"}}, "required": []}
}
for function in functions # This assumes 'functions' is globally accessible
for function in functions
]
class OpenAIProvider(AIProvider):
@@ -72,7 +74,7 @@ class OpenAIProvider(AIProvider):
response = self.client.chat.completions.create(
model=self.model,
messages=self.format_messages(messages),
functions=functions, # This assumes 'functions' is globally accessible
functions=functions,
function_call="auto",
max_tokens=self.get_max_tokens()
)