Add interface for listing LLM models.

This commit is contained in:
Bartosz Taudul
2025-05-11 17:24:33 +02:00
parent cb212567ed
commit 08c877c4ef
2 changed files with 8 additions and 0 deletions

View File

@@ -26,4 +26,9 @@ std::string TracyLlm::GetVersion() const
return m_ollama->get_version();
}
std::vector<std::string> TracyLlm::GetModels() const
{
return m_ollama->list_models();
}
}

View File

@@ -2,6 +2,8 @@
#define __TRACYLLM_HPP__
#include <memory>
#include <string>
#include <vector>
class Ollama;
@@ -16,6 +18,7 @@ public:
[[nodiscard]] bool IsValid() const { return m_valid; }
[[nodiscard]] std::string GetVersion() const;
[[nodiscard]] std::vector<std::string> GetModels() const;
private:
std::unique_ptr<Ollama> m_ollama;