| iomodo | 7554232 | 2025-07-30 19:27:48 +0400 | [diff] [blame] | 1 | package provider |
| 2 | |||||
| 3 | import ( | ||||
| 4 | "github.com/iomodo/staff/llm" | ||||
| 5 | "github.com/iomodo/staff/llm/fake" | ||||
| 6 | "github.com/iomodo/staff/llm/openai" | ||||
| 7 | ) | ||||
| 8 | |||||
| 9 | func CreateProvider(config llm.Config) llm.LLMProvider { | ||||
| 10 | switch config.Provider { | ||||
| 11 | case llm.ProviderFake: | ||||
| 12 | return fake.New() | ||||
| 13 | case llm.ProviderOpenAI: | ||||
| 14 | return openai.New(config) | ||||
| 15 | default: | ||||
| 16 | return fake.New() | ||||
| 17 | } | ||||
| 18 | |||||
| 19 | } | ||||