LordXido commited on
Commit
b4dc5ac
·
verified ·
1 Parent(s): 072654d

Create llm_sim.py

Browse files
Files changed (1) hide show
  1. llm_sim.py +32 -0
llm_sim.py ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Toy LLM plug-ins for offline demos.
3
+ Each provider exposes a .generate(prompt) method.
4
+ Extend or swap with real API wrappers later.
5
+ """
6
+
7
+ class BaseToyLLM:
8
+ name = "base"
9
+ def generate(self, prompt: str) -> str:
10
+ return f"[{self.name}] echo: {prompt[:60]}..."
11
+
12
+ class GPTToy(BaseToyLLM):
13
+ name = "gpt-toy"
14
+ def generate(self, prompt): # pretend “smart”
15
+ return "[GPT-Toy] ➜ I understand and comply."
16
+
17
+ class ClaudeToy(BaseToyLLM):
18
+ name = "claude-toy"
19
+ def generate(self, prompt):
20
+ return "[Claude-Toy] ➜ Here’s a thoughtful answer."
21
+
22
+ class MistralToy(BaseToyLLM):
23
+ name = "mistral-toy"
24
+ def generate(self, prompt):
25
+ return "[Mistral-Toy] ➜ Crisp, minimal reply."
26
+
27
+ # Registry of available toy providers
28
+ LLM_REGISTRY = {
29
+ "gpt": GPTToy(),
30
+ "claude": ClaudeToy(),
31
+ "mistral": MistralToy(),
32
+ }