project_settings.cpp 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. #include "project_settings.hpp"
  2. #include "collection_registry.hpp"
  3. #include "settings.hpp"
  4. namespace svapi {
  5. ProjectEmbeddingSettings ProjectEmbeddingSettings::fromJson(const nlohmann::json& j) {
  6. ProjectEmbeddingSettings p;
  7. p.project = j.value("project", p.project);
  8. p.openaiApiBase = j.value("openai_api_base", p.openaiApiBase);
  9. p.openaiApiKey = j.value("openai_api_key", p.openaiApiKey);
  10. p.defaultEmbeddingModel = j.value("default_embedding_model", p.defaultEmbeddingModel);
  11. return p;
  12. }
  13. nlohmann::json ProjectEmbeddingSettings::toJson() const {
  14. return {{"project", project},
  15. {"openai_api_base", openaiApiBase},
  16. {"openai_api_key", openaiApiKey},
  17. {"default_embedding_model", defaultEmbeddingModel}};
  18. }
  19. std::string projectSettingsDocId(const std::string& project) { return "project_" + project; }
  20. ResolvedEmbedding resolveEmbedding(const Settings& global, const ProjectEmbeddingSettings* proj,
  21. const CollectionMeta& coll) {
  22. ResolvedEmbedding r{global.openaiApiBase, global.openaiApiKey, global.defaultEmbeddingModel};
  23. if (proj) {
  24. if (!proj->openaiApiBase.empty()) r.apiBase = proj->openaiApiBase;
  25. if (!proj->openaiApiKey.empty()) r.apiKey = proj->openaiApiKey;
  26. if (!proj->defaultEmbeddingModel.empty()) r.model = proj->defaultEmbeddingModel;
  27. }
  28. if (!coll.embeddingModel.empty()) r.model = coll.embeddingModel;
  29. return r;
  30. }
  31. } // namespace svapi