Upload 5 files

skills/control_mode.py

@@ -0,0 +1,22 @@
+import requests
+
+
+
+def manual_control_mode(command):
+    url = "https://srivatsavdamaraju-central-robot-server.hf.space/activate_manual_mode"
+    payload = {}
+    headers = {}
+    response = requests.request("POST", url, headers=headers, data=payload)
+    print(response.text)
+    return "this is manual control mode"
+
+
+def deactivate_manual_control_mode(command):
+    url = (
+        "https://srivatsavdamaraju-central-robot-server.hf.space/deactivate_manual_mode"
+    )
+    payload = {}
+    headers = {}
+    response = requests.request("POST", url, headers=headers, data=payload)
+    print(response.text)
+    return "this is deActivate manual control mode"

skills/serch.py

@@ -0,0 +1,8 @@
+# /skills/search_tool.py
+
+from langchain.tools import DuckDuckGoSearchRun
+
+search = DuckDuckGoSearchRun()
+
+def search_tool(query):
+    return search.run(query)

skills/utils.py

@@ -0,0 +1,5 @@
+def fix_base64_padding(base64_string: str) -> str:
+    padding_needed = len(base64_string) % 4
+    if padding_needed:
+        base64_string += '=' * (4 - padding_needed)
+    return base64_string

skills/vision.py

@@ -0,0 +1,38 @@
+# /skills/image_analysis.py
+
+import base64
+from groq import Groq
+
+# Function to encode the image
+def encode_image(image_file):
+    return base64.b64encode(image_file.read()).decode('utf-8')
+
+def get_chat_completion(query, image_file=None):
+    client = Groq()
+
+    messages = [
+        {
+            "role": "user",
+            "content": [
+                {"type": "text", "text": query},
+            ],
+        }
+    ]
+
+    if image_file:
+        base64_image = encode_image(image_file)
+        messages[0]["content"].append(
+            {
+                "type": "image_url",
+                "image_url": {
+                    "url": f"data:image/jpeg;base64,{base64_image}",
+                },
+            }
+        )
+
+    chat_completion = client.chat.completions.create(
+        messages=messages,
+        model="llama-3.2-11b-vision-preview",
+    )
+
+    return chat_completion.choices[0].message.content

skills/wiki.py

@@ -0,0 +1,9 @@
+# /skills/wikipedia_tool.py
+
+from langchain.tools import WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
+
+wikipedia = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
+
+def wikipedia_tool(query):
+    return wikipedia.run(query)