diff --git a/pkg/ai/iai.go b/pkg/ai/iai.go
index 15e1b65d..0d2d9821 100644
--- a/pkg/ai/iai.go
+++ b/pkg/ai/iai.go
@@ -78,7 +78,17 @@ type IAIConfig interface {
 	GetTemperature() float32
 	GetTopP() float32
 	GetMaxTokens() int
-	GetConfigName() string // Added to support multiple configurations
+	GetConfigName() string
+	GetPassword() string
+	GetBaseURL() string
+	GetProxyEndpoint() string
+	GetEndpointName() string
+	GetEngine() string
+	GetProviderId() string
+	GetCompartmentId() string
+	GetOrganizationId() string
+	GetCustomHeaders() []http.Header
+	GetTopK() int32
 }
 
 func NewClient(provider string) IAI {
diff --git a/pkg/ai/openai_header_transport_test.go b/pkg/ai/openai_header_transport_test.go
index 9d43f463..76ac03cd 100644
--- a/pkg/ai/openai_header_transport_test.go
+++ b/pkg/ai/openai_header_transport_test.go
@@ -76,6 +76,10 @@ func (m *mockConfig) GetProviderRegion() string {
 	return ""
 }
 
+func (m *mockConfig) GetConfigName() string {
+	return ""
+}
+
 func TestOpenAIClient_CustomHeaders(t *testing.T) {
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		assert.Equal(t, "Value1", r.Header.Get("X-Custom-Header-1"))