Skip to content

Commit fdc8e8e

Browse files
Added data
1 parent 186ebe6 commit fdc8e8e

File tree

6 files changed

+23
-9
lines changed

6 files changed

+23
-9
lines changed

data/exp_raw/CommonDataModel

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit 1e0360980a6df4fee8e94e52c20576b21a0803c9

data/exp_raw/FunSQL.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit b3a400f2bc84b60b63949c8a08fa7dc442e0b28b
Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Subproject commit 637960b01f08ec3ce6e66c73d74bddf8556dde2e

scripts/load_model.jl

Lines changed: 0 additions & 8 deletions
This file was deleted.

src/JuliaHealthLLM.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ module JuliaHealthLLM
33
using LibGit2
44
using Downloads
55
using Dates
6+
using PromptingTools
67
using TextAnalysis
78
using Serialization, Base.Threads
89
using Transformers, CUDA
@@ -15,7 +16,6 @@ include("../scripts/knowledge.jl")
1516
include("../scripts/restore_submodules.jl")
1617
include("../scripts/update_julia_package_repos.jl")
1718
include("../scripts/update_and_record.jl")
18-
include("../scripts/load_model.jl")
1919
include("../src/chunk_docs.jl")
2020

2121
export clone_repositories, normalize_repo_url, clone_repo, is_textual_file,

src/rag.jl

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
const PT = PromptingTools
2+
3+
function load_model_ollama(model_name::String)
4+
PT.register_model!(name=model_name, schema = PT.OllamaSchema())
5+
end
6+
7+
function load_embedding_ollama(model_name::String)
8+
PT.register_model!(name=model_name, schema = PT.OllamaSchema())
9+
end
10+
11+
function embed_docs(model_name::String, docs::Vector{String})
12+
load_embedding_ollama(model_name)
13+
end
14+
15+
function convo(message::String, model_name::String)
16+
load_model_ollama(model_name)
17+
msg = aimessage(message, model_name)
18+
return msg
19+
end

0 commit comments

Comments
 (0)