scriptos/ollamarama-matrix-chatbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

moved api base url into config

Browse Source
This commit is contained in:
Dustin 2024-05-15 20:52:37 -04:00
parent 000b2b154c
commit cd68cdf93c
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
config.json
View File

@ -37,5 +37,8 @@
"admin_nick2" "admin_nick2"
] ]
},
{
"api_base": "http://localhost:11434"
} }
] ]

5
ollamarama.py
View File

@ -19,7 +19,8 @@ class ollamarama:
config = json.load(f) config = json.load(f)
f.close() f.close()
self.server, self.username, self.password, self.channels, self.default_personality, self.admins = config[1].values() self.server, self.username, self.password, self.channels, self.default_personality, self.admins = config[1].values()
self.api_base = config[2]['api_base']
self.personality = self.default_personality self.personality = self.default_personality
self.client = AsyncClient(self.server, self.username) self.client = AsyncClient(self.server, self.username)
@ -93,7 +94,7 @@ class ollamarama:
try: try:
#Generate response #Generate response
response = completion( response = completion(
api_base="http://localhost:11434", api_base=self.api_base,
model=self.model, model=self.model,
temperature=self.temperature, temperature=self.temperature,
top_p=self.top_p, top_p=self.top_p,