[#106] updated provider and added model struct (#109)

* [#genai] updated model id

* [#genai] updated models method signature

Author: nandagopalan

genai/provider.go

@@ -1,8 +1,6 @@
 package genai
 
 import (
-	"io"
-
 	"oss.nandlabs.io/golly/errutils"
 	"oss.nandlabs.io/golly/managers"
 )
@@ -34,6 +32,24 @@ var GetUnsupportedProviderErr = errutils.NewCustomError("unsupported provider fo
 
 var Providers managers.ItemManager[Provider] = managers.NewItemManager[Provider]()
 
+type StreamingHandller func(message Message, last bool)
+
+type Model struct {
+	Id       string         `json:"id" yaml:"id" bson:"id"`
+	Name     string         `json:"name" yaml:"name" bson:"name"`
+	Version  string         `json:"version" yaml:"version" bson:"version"`
+	InMimes  []string       `json:"in_mimes" yaml:"in_mimes" bson:"in_mimes"`
+	OutMimes []string       `json:"out_mimes" yaml:"out_mimes" bson:"out_mimes"`
+	Options  []*ModelOption `json:"options" yaml:"options" bson:"options"`
+}
+
+type ModelOption struct {
+	Name     string `json:"name" yaml:"name" bson:"name"`
+	Type     string `json:"type" yaml:"type" bson:"type"`
+	Required bool   `json:"required" yaml:"required" bson:"required"`
+	Default  any    `json:"default" yaml:"default" bson:"default"`
+}
+
 // Provider is the interface that represents a generative AI model
 type Provider interface {
 	// Name returns the name of the model
@@ -42,20 +58,18 @@ type Provider interface {
 	Description() string
 	// Version returns the version of the model
 	Version() string
-	// Author returns the author of the model
-	Author() string
-	// License returns the license of the model
-	License() string
-	// Supports returns true if the model supports the given MIME type as a consumer or provider
-	Supports(model, mime string) (consumer bool, provider bool)
-	// Accepts returns the supported input MIME types accepted by this model
-	Accepts(model string) []string
-	// Produces returns the supported output MIME types thats produced by this model
-	Produces(model string) []string
-	// Generate will invoke the model generation and fetch the result. This is a blocking call.
+	//Models returns the list of models supported by the provider
+	// Some providers may not have to get this information programatically, in which case this method should return nil
+	// and unsupported operation error.
+	Models() ([]*Model, error)
+	// Generate will invoke the model generation and adds the result to the exhange. This is a blocking call.
 	Generate(model string, exchange Exchange, options *Options) error
-	// GenerateStream will invoke the model generation and stream the result. This will be a non-blocking call.
-	GenerateStream(model string, exchange Exchange, handler func(reader io.Reader), options Options) error
+	// GenerateStream will invoke the model generation and stream the result to handler function.
+	// This will be a non-blocking call.
+	// The handler function will be called for each message generated by the model.
+	// The last parameter will be true for the last message.
+	// The exchange will also be updated with the generated messages.
+	GenerateStream(model string, exchange Exchange, handler StreamingHandller, options Options) error
 }
 
 // Options represents the options for the Service
@@ -686,86 +700,3 @@ func (o *OptionsBuilder) SetOutputMimes(outputMimes ...string) *OptionsBuilder {
 	o.options.values[OptionOutputMimes] = outputMimes
 	return o
 }
-
-// AbstractModel represents a generic model with metadata information.
-// It includes fields for the model's name, description, version, author, license,
-// and supported input and output MIME types.
-type AbstractModel struct {
-	name        string
-	description string
-	version     string
-	author      string
-	license     string
-	inputMime   []string
-	outputMime  []string
-}
-
-// Name returns the name of the AbstractModel.
-// It retrieves the value of the private field 'name' and returns it as a string.
-//
-// Returns:
-//
-//	string: The name of the model.
-func (m *AbstractModel) Name() string {
-	return m.name
-}
-
-// Description returns the description of the AbstractModel.
-// It provides a brief summary or details about the model.
-//
-// Returns:
-//
-//	string: The description of the model.
-func (m *AbstractModel) Description() string {
-	return m.description
-}
-
-// Version returns the version of the AbstractModel.
-// It retrieves the value of the private field 'version' and returns it as a string.
-//
-// Returns:
-//
-//	string: The version of the model.
-func (m *AbstractModel) Version() string {
-	return m.version
-}
-
-// Author returns the author of the AbstractModel.
-// It retrieves the value of the private field 'author' and returns it as a string.
-//
-// Returns:
-//
-//	string: The author of the model.
-func (m *AbstractModel) Author() string {
-	return m.author
-}
-
-// License returns the license of the AbstractModel.
-// It retrieves the value of the private field 'license' and returns it as a string.
-//
-// Returns:
-//
-//	string: The license of the model.
-func (m *AbstractModel) License() string {
-	return m.license
-}
-
-// InputMimeTypes returns the supported input MIME types for the AbstractModel.
-// It retrieves the value of the private field 'inputMime' and returns it as a slice of strings.
-//
-// Returns:
-//
-//	[]string: A slice of strings representing the supported input MIME types.
-func (m *AbstractModel) InputMimeTypes() []string {
-	return m.inputMime
-}
-
-// OutputMimeTypes returns the supported output MIME types for the AbstractModel.
-// It retrieves the value of the private field 'outputMime' and returns it as a slice of strings.
-//
-// Returns:
-//
-//	[]string: A slice of strings representing the supported output MIME types.
-func (m *AbstractModel) OutputMimeTypes() []string {
-	return m.outputMime
-}