Add GoogleOpenAISchema with OpenAI compatibility mode

- Add GoogleOpenAISchema struct with documentation
- Implement create_chat method with Google API base URL
- Implement create_embedding method with Google API base URL
- Use GOOGLE_API_KEY for authentication

src/llm_interface.jl

@@ -274,6 +274,23 @@ Requires one environment variable to be set:
 """
 struct XAIOpenAISchema <: AbstractOpenAISchema end
 
+"""
+    GoogleOpenAISchema
+
+Schema to call the Google's Gemini API using OpenAI compatibility mode. [API Reference](https://ai.google.dev/gemini-api/docs/openai#rest)
+
+Links:
+- [Get your API key](https://aistudio.google.com/apikey)
+- [API Reference](https://ai.google.dev/gemini-api/docs/openai#rest)
+- [Available models](https://ai.google.dev/models/gemini)
+
+Requires one environment variable to be set:
+- `GOOGLE_API_KEY`: Your API key
+
+The base URL for the API is "https://generativelanguage.googleapis.com/v1beta"
+"""
+struct GoogleOpenAISchema <: AbstractOpenAISchema end
+
 abstract type AbstractOllamaSchema <: AbstractPromptSchema end
 
 """
@@ -517,4 +534,4 @@ end
 abstract type AbstractExtractedData end
 Base.show(io::IO, x::AbstractExtractedData) = dump(io, x; maxdepth = 1)
 "Check if the object is an instance of `AbstractExtractedData`"
-isextracted(x) = x isa AbstractExtractedData
+isextracted(x) = x isa AbstractExtractedData

src/llm_openai_schema_defs.jl

@@ -214,6 +214,15 @@ function OpenAI.create_chat(schema::XAIOpenAISchema,
     api_key = isempty(XAI_API_KEY) ? api_key : XAI_API_KEY
     OpenAI.create_chat(CustomOpenAISchema(), api_key, model, conversation; url, kwargs...)
 end
+function OpenAI.create_chat(schema::GoogleOpenAISchema,
+        api_key::AbstractString,
+        model::AbstractString,
+        conversation;
+        url::String = "https://generativelanguage.googleapis.com/v1beta",
+        kwargs...)
+    api_key = isempty(GOOGLE_API_KEY) ? api_key : GOOGLE_API_KEY
+    OpenAI.create_chat(CustomOpenAISchema(), api_key, model, conversation; url, kwargs...)
+end
 function OpenAI.create_chat(schema::DatabricksOpenAISchema,
         api_key::AbstractString,
         model::AbstractString,
@@ -384,6 +393,17 @@ function OpenAI.create_embeddings(schema::XAIOpenAISchema,
         base_url = url)
     OpenAI.create_embeddings(provider, docs, model; kwargs...)
 end
+function OpenAI.create_embeddings(schema::GoogleOpenAISchema,
+        api_key::AbstractString,
+        docs,
+        model::AbstractString;
+        url::String = "https://generativelanguage.googleapis.com/v1beta",
+        kwargs...)
+    provider = CustomProvider(;
+        api_key = isempty(GOOGLE_API_KEY) ? api_key : GOOGLE_API_KEY,
+        base_url = url)
+    OpenAI.create_embeddings(provider, docs, model; kwargs...)
+end
 function OpenAI.create_embeddings(schema::AzureOpenAISchema,
         api_key::AbstractString,
         docs,
@@ -441,4 +461,4 @@ function OpenAI.create_images(schema::TestEchoOpenAISchema,
     schema.model_id = get(kwargs, :model, "")
     schema.inputs = prompt
     return schema
-end
+end