diff --git a/profiler/src/profiler/TracyLlm.cpp b/profiler/src/profiler/TracyLlm.cpp index 1c86f943..70562158 100644 --- a/profiler/src/profiler/TracyLlm.cpp +++ b/profiler/src/profiler/TracyLlm.cpp @@ -26,4 +26,9 @@ std::string TracyLlm::GetVersion() const return m_ollama->get_version(); } +std::vector TracyLlm::GetModels() const +{ + return m_ollama->list_models(); +} + } diff --git a/profiler/src/profiler/TracyLlm.hpp b/profiler/src/profiler/TracyLlm.hpp index 30bd749e..d9a8f818 100644 --- a/profiler/src/profiler/TracyLlm.hpp +++ b/profiler/src/profiler/TracyLlm.hpp @@ -2,6 +2,8 @@ #define __TRACYLLM_HPP__ #include +#include +#include class Ollama; @@ -16,6 +18,7 @@ public: [[nodiscard]] bool IsValid() const { return m_valid; } [[nodiscard]] std::string GetVersion() const; + [[nodiscard]] std::vector GetModels() const; private: std::unique_ptr m_ollama;