config

type Category struct {
	ID     string `json:"id"`
	Name   string `json:"name"`
	NameEn string `json:"nameEn,omitempty"`
}

type Config struct {
	// SystemDatasource specifies the built-in datasource for Arcadia to host data files and model files
	SystemDatasource arcadiav1alpha1.TypedObjectReference `json:"systemDatasource,omitempty"`

	// RelationalDatasource specifies the built-in datasource(common:postgres) for Arcadia to host relational data
	RelationalDatasource arcadiav1alpha1.TypedObjectReference `json:"relationalDatasource,omitempty"`

	// Gateway to access LLM api services
	Gateway *Gateway `json:"gateway,omitempty"`

	// EmbeddingSuite here represents the system embedding service provided by the system
	EmbeddingSuite

	// Resource pool managed by Ray cluster
	RayClusters []RayCluster `json:"rayClusters,omitempty"`

	// the default rerank model
	Rerank *arcadiav1alpha1.TypedObjectReference `json:"rerank,omitempty"`

	// Streamlit to get the Streamlit configuration
	// Deprecated: this field no longer maintained
	Streamlit *Streamlit `json:"streamlit,omitempty"`
}

type EmbeddingSuite struct {
	// Embedder specifies the default embedder for Arcadia to generate embeddings
	Embedder *arcadiav1alpha1.TypedObjectReference `json:"embedder,omitempty"`

	// VectorStore to access VectorStore api services
	VectorStore *arcadiav1alpha1.TypedObjectReference `json:"vectorStore,omitempty"`
}

type GPTsConfig struct {
	// URL is the url of gpt store
	URL string `json:"url,omitempty"`
	// PublicNamespace is the namespace which all gpt-releated resources are public
	PublicNamespace string     `json:"public_namespace,omitempty"`
	Categories      []Category `json:"categories,omitempty"`
}

type Gateway struct {
	// ExternalAPIServer is the api(LLM/Embedding) server address that can be accessed from internet
	ExternalAPIServer string `json:"externalApiServer,omitempty"`
	// APIServer is api(LLM/Embedding) server which can be accessed within platform
	APIServer string `json:"apiServer,omitempty"`
	// Controller is the server address which is responsible for llm/embedding service registration
	Controller string `json:"controller,omitempty"`
}

type RayCluster struct {
	// Name of this ray cluster
	Name string `json:"name,omitempty"`
	// Address of ray head address
	HeadAddress string `json:"headAddress,omitempty"`
	// Management dashboard of ray cluster, optional to configure it using ingress
	DashboardHost string `json:"dashboardHost,omitempty"`
	// Overwrite the python version in the woker
	PythonVersion string `json:"pythonVersion,omitempty"`
	// Ray cluster version
	RayVersion string `json:"rayVersion,omitempty"`
}

type Streamlit struct {
	Image            string `json:"image"`
	IngressClassName string `json:"ingressClassName"`
	Host             string `json:"host"`
	ContextPath      string `json:"contextPath"`
}